Completes OPEN-3570 Add backend support for baseline models

Author: gustavocidornelas

openlayer/__init__.py

@@ -451,14 +451,6 @@ def add_baseline_model(
 
                 For baseline models, the content of the YAML file should contain:
 
-                - ``ensembleSize`` : int, default 10
-                    Number of models ensembled.
-                - ``randomSeed`` : int, default 42
-                    Random seed to be used for model training.
-                - ``timeout`` : int, default 60
-                    Maximum time (in seconds) to train all the models.
-                - ``perRunLimit`` : int, optional
-                    Maximum time (in seconds) to train each model.
                 - ``metadata`` : Dict[str, any], default {}
                     Dictionary containing metadata about the model. This is the
                     metadata that will be displayed on the Openlayer platform.

openlayer/projects.py

@@ -51,7 +51,7 @@ def add_baseline_model(
         **kwargs,
     ):
         """Adds a baseline model to the project."""
-        return self.client.add_baseline(
+        return self.client.add_baseline_model(
             *args, project_id=self.id, task_type=tasks.TaskType(self.taskType), **kwargs
         )
 

openlayer/schemas.py

@@ -21,11 +21,7 @@
 class BaselineModelSchema(ma.Schema):
     """Schema for baseline models."""
 
-    ensembleSize = ma.fields.Int(load_default=10)
     metadata = ma.fields.Dict(allow_none=True, load_default={})
-    perRunLimit = ma.fields.Int(load_default=None, allow_none=True)
-    randomSeed = ma.fields.Int(load_default=42)
-    timeout = ma.fields.Int(load_default=60)
 
 
 class CommitSchema(ma.Schema):