Remove LLM task type short-circuit from Python client methods

openlayer-ai · Aug 22, 2023 · f155af5 · f155af5
1 parent cbe83fe
commit f155af5
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/openlayer/__init__.py b/openlayer/__init__.py
@@ -432,10 +432,6 @@ def add_model(
                     f"{len(sample_data)} rows were provided."
                 )
 
-        # hack ---------------------------------------------------------------
-        task_type = TaskType.LLM
-        # hack ---------------------------------------------------------------
-
         # Validate model package
         model_validator = model_validators.get_validator(
             task_type=task_type,
@@ -751,10 +747,6 @@ class probabilities. For example, for a binary classification
         >>> project.commit("Initial dataset commit.")
         >>> project.push()
         """
-        # hack ---------------------------------------------------------------
-        task_type = TaskType.LLM
-        # hack ---------------------------------------------------------------
-
         # Validate dataset
         dataset_validator = dataset_validators.get_validator(
             task_type=task_type,
@@ -1113,10 +1105,6 @@ def push(self, project_id: str, task_type: TaskType) -> Optional[ProjectVersion]
         """
         project_dir = f"{constants.OPENLAYER_DIR}/{project_id}/staging"
 
-        # hack ---------------------------------------------------------------
-        task_type = TaskType.LLM
-        # hack ---------------------------------------------------------------
-
         if self._ready_for_push(project_dir=project_dir, task_type=task_type):
             with open(
                 f"{project_dir}/commit.yaml", "r", encoding="UTF-8"