Fix pylint issues

Author: gustavocidornelas

openlayer/__init__.py

@@ -1183,7 +1183,7 @@ def _post_push_cleanup(self, project_dir: str) -> None:
         shutil.rmtree(project_dir)
         os.makedirs(project_dir, exist_ok=True)
 
-    def export(self, destination_dir: str, project_id: int):
+    def export(self, destination_dir: str, project_id: int, task_type: TaskType):
         """Exports the commited resources as a tarfile to the location specified
         by ``destination_dir``.
 
@@ -1216,7 +1216,7 @@ def export(self, destination_dir: str, project_id: int):
         """
         project_dir = f"{OPENLAYER_DIR}/{project_id}/staging"
 
-        if self._ready_for_push(project_dir=project_dir):
+        if self._ready_for_push(project_dir=project_dir, task_type=task_type):
             # Tar the project's staging area
             with tempfile.TemporaryDirectory() as tmp_dir:
                 tar_file_path = os.path.join(tmp_dir, "tarfile")

openlayer/models.py

@@ -467,10 +467,10 @@ def get_model_runner(
     logger : Optional[logging.Logger], optional
         Logger to use, by default None
     """
-    if (
-        task_type == tasks.TaskType.TabularClassification
-        or task_type == tasks.TaskType.TextClassification
-    ):
+    if task_type in [
+        tasks.TaskType.TabularClassification,
+        tasks.TaskType.TextClassification,
+    ]:
         return ClassificationModelRunner(model_package, logger)
     elif task_type == tasks.TaskType.TabularRegression:
         return RegressionModelRunner(model_package, logger)

openlayer/projects.py

@@ -85,7 +85,9 @@ def push(self, *args, **kwargs):
 
     def export(self, *args, **kwargs):
         """Exports the commited resources to a specified location."""
-        return self.client.export(*args, project_id=self.id, **kwargs)
+        return self.client.export(
+            *args, project_id=self.id, task_type=tasks.TaskType(self.taskType), **kwargs
+        )
 
     def status(self, *args, **kwargs):
         """Shows the state of the staging area."""

openlayer/schemas.py

@@ -53,7 +53,7 @@ class BaselineModelSchema(maos.OneOfSchema):
 
     def get_obj_type(self, obj):
         if obj != "tabular-classification":
-            raise Exception(f"Unknown object type: {obj.__class__.__name__}")
+            raise ma.ValidationError(f"Unknown object type: {obj.__class__.__name__}")
         return obj
 
 
@@ -187,7 +187,7 @@ def get_obj_type(self, obj):
             "text-classification",
             "tabular-regression",
         }:
-            raise Exception(f"Unknown object type: {obj.__class__.__name__}")
+            raise ma.ValidationError(f"Unknown object type: {obj.__class__.__name__}")
         return obj
 
 
@@ -299,7 +299,7 @@ def get_obj_type(self, obj):
             "text-classification",
             "tabular-regression",
         }:
-            raise Exception(f"Unknown object type: {obj.__class__.__name__}")
+            raise ma.ValidationError(f"Unknown object type: {obj.__class__.__name__}")
         return obj