Fix pylint issues and bump version

Author: gustavocidornelas

openlayer/__init__.py

@@ -40,8 +40,6 @@
 from .projects import Project
 from .schemas import BaselineModelSchema, DatasetSchema, ModelSchema
 from .tasks import TaskType
-
-# from validators import models as model_validators
 from .validators import (
     baseline_model_validators,
     commit_validators,

openlayer/inference_pipelines.py

@@ -1,6 +1,5 @@
 """Module for the InferencePipeline class.
 """
-from . import tasks
 
 
 class InferencePipeline:
@@ -13,6 +12,7 @@ def __init__(self, json, upload, client, task_type):
         self.project_id = json["projectId"]
         self.upload = upload
         self.client = client
+        # pylint: disable=invalid-name
         self.taskType = task_type
 
     def __getattr__(self, name):

openlayer/projects.py

@@ -54,7 +54,8 @@ def add_model(
 
         - A **full model** is comprised of metadata and artifacts. Thus, the model config, model package, and sample data arguments are needed.
 
-        - A **direct-to-API model** applies only to **LLMs**. Only the model config argument is required, but the model config must contain all the information needed to make predictions -- such as the ``prompt``, the ``modelProvider``, and others.
+        - A **direct-to-API model** applies only to **LLMs**. Only the model config argument is required, but the model
+            config must contain all the information needed to make predictions -- such as the ``prompt``, the ``modelProvider``, and others.
 
         Parameters
         ----------
@@ -449,7 +450,8 @@ def commit(self, *args, **kwargs):
 
         Notes
         -----
-        - To use this method, you must first add a model and/or dataset to the staging area using one of the ``add_*`` methods (e.g., :obj:`add_model`, :obj:`add_dataset`, :obj:`add_dataframe`).
+        - To use this method, you must first add a model and/or dataset to the staging
+            area using one of the ``add_*`` methods (e.g., :obj:`add_model`, :obj:`add_dataset`, :obj:`add_dataframe`).
 
         Examples
         --------

openlayer/schemas.py

@@ -555,7 +555,7 @@ class ReferenceDatasetSchema(maos.OneOfSchema):
     Returns the correct schema based on the task type."""
 
     type_field = "task_type"
-    # pylint: ignore=line-too-long
+    # pylint: disable=line-too-long
     type_schemas = {
         TaskType.TabularClassification.value: TabularClassificationReferenceDatasetSchema,
         TaskType.TabularRegression.value: TabularRegressionReferenceDatasetSchema,

openlayer/validators/dataset_validators.py

@@ -1,3 +1,4 @@
+# pylint: disable=bare-except
 """Implements the dataset specific validation classes.
 """
 import ast
@@ -125,7 +126,8 @@ def _load_dataset_config(self):
                     self.dataset_config_file_path, "r", encoding="UTF-8"
                 ) as stream:
                     self.dataset_config = yaml.safe_load(stream)
-            except:
+            # pylint: disable=broad-exception-caught
+            except Exception:
                 self.failed_validations.append(
                     f"File `{self.dataset_config_file_path}` is not a valid .yaml file."
                 )
@@ -255,7 +257,8 @@ def _timestamps_are_unix_epoch_seconds(
         try:
             # Note the unit="s" argument
             pd.to_datetime(dataset_df[timestamp_column_name], unit="s")
-        except:
+        # pylint: disable=broad-exception-caught
+        except Exception:
             return False
         return True
 

openlayer/version.py

@@ -22,4 +22,4 @@
          data=data,
       )
 """
-__version__ = "0.1.0a9"
+__version__ = "0.1.0a10"