ml-management 0.6.3rc11__tar.gz → 0.6.3rc12__tar.gz

ml_management-0.6.3rc12/ML_management/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.6.3rc12"

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/ML_management/graphql/schema.py

@@ -1562,7 +1562,7 @@ class ModelVersionPagination(sgqlc.types.Type):
 
 class Mutation(sgqlc.types.Type):
     __schema__ = schema
-    __field_names__ = ('add_ml_job', 'cancel_build_job_for_executor_version', 'cancel_build_job_for_model_version', 'cancel_job', 'cancel_venv_build_job_for_model_version', 'create_local_job', 'delete_dataset_loader', 'delete_dataset_loader_tag', 'delete_dataset_loader_version_from_name_version', 'delete_dataset_loader_version_tag', 'delete_executor', 'delete_executor_tag', 'delete_executor_version_from_name_version', 'delete_executor_version_tag', 'delete_experiment_tag', 'delete_model', 'delete_model_tag', 'delete_model_version_from_name_version', 'delete_model_version_tag', 'log_metric', 'log_metrics', 'log_param', 'log_params', 'rebuild_model_version_image', 'rebuild_no_model_executor_version_image', 'rename_experiment', 'serve_model', 'set_dataset_loader_tag', 'set_dataset_loader_version_tag', 'set_executor_tag', 'set_executor_version_tag', 'set_experiment_tag', 'set_model_tag', 'set_model_version_tag', 'start_local_job', 'stop_local_job', 'stop_model_serving', 'update_dataset_loader', 'update_dataset_loader_version', 'update_executor', 'update_executor_version', 'update_experiment', 'update_job', 'update_model', 'update_model_version')
+    __field_names__ = ('add_ml_job', 'cancel_build_job_for_executor_version', 'cancel_build_job_for_model_version', 'cancel_job', 'cancel_venv_build_job_for_model_version', 'create_experiment', 'create_local_job', 'delete_dataset_loader', 'delete_dataset_loader_tag', 'delete_dataset_loader_version_from_name_version', 'delete_dataset_loader_version_tag', 'delete_executor', 'delete_executor_tag', 'delete_executor_version_from_name_version', 'delete_executor_version_tag', 'delete_experiment_tag', 'delete_model', 'delete_model_tag', 'delete_model_version_from_name_version', 'delete_model_version_tag', 'log_metric', 'log_metrics', 'log_param', 'log_params', 'rebuild_model_version_image', 'rebuild_no_model_executor_version_image', 'rename_experiment', 'serve_model', 'set_dataset_loader_tag', 'set_dataset_loader_version_tag', 'set_executor_tag', 'set_executor_version_tag', 'set_experiment_tag', 'set_model_tag', 'set_model_version_tag', 'start_local_job', 'stop_local_job', 'stop_model_serving', 'update_dataset_loader', 'update_dataset_loader_version', 'update_executor', 'update_executor_version', 'update_experiment', 'update_job', 'update_model', 'update_model_version')
     add_ml_job = sgqlc.types.Field(sgqlc.types.non_null(ExecutionJob), graphql_name='addMlJob', args=sgqlc.types.ArgDict((
         ('form', sgqlc.types.Arg(sgqlc.types.non_null(JobParameters), graphql_name='form', default=None)),
 ))
@@ -1614,6 +1614,19 @@ class Mutation(sgqlc.types.Type):
     * `version` (`Int!`)None
     '''
 
+    create_experiment = sgqlc.types.Field(sgqlc.types.non_null(Experiment), graphql_name='createExperiment', args=sgqlc.types.ArgDict((
+        ('experiment_description', sgqlc.types.Arg(String, graphql_name='experimentDescription', default='')),
+        ('experiment_name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='experimentName', default=None)),
+        ('visibility', sgqlc.types.Arg(sgqlc.types.non_null(VisibilityOptions), graphql_name='visibility', default='PRIVATE')),
+))
+    )
+    '''Arguments:
+
+    * `experiment_description` (`String`)None (default: `""`)
+    * `experiment_name` (`String!`)None
+    * `visibility` (`VisibilityOptions!`)None (default: `PRIVATE`)
+    '''
+
     create_local_job = sgqlc.types.Field(sgqlc.types.non_null(ExecutionJob), graphql_name='createLocalJob', args=sgqlc.types.ArgDict((
         ('experiment_name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='experimentName', default=None)),
         ('job_name', sgqlc.types.Arg(String, graphql_name='jobName', default=None)),

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/ML_management/mlmanagement/log_api.py

@@ -147,8 +147,8 @@ def _log_object_src(
     finally:
         sys.path = old_python_path
         for module in set(sys.modules) - set(old_sys_modules):
-            if hasattr(sys.modules[module], __file__):
-                if str(Path(*parts[:-2])) in sys.modules[module].__file__:
+            if hasattr(sys.modules[module], "__file__"):
+                if sys.modules[module].__file__ and str(Path(*parts[:-2])) in sys.modules[module].__file__:
                     sys.modules.pop(module)
 
     if additional_local_packages:

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/ML_management/sdk/__init__.py

@@ -41,6 +41,7 @@ set_experiment_description = experiment.set_experiment_description
 delete_experiment_tag = experiment.delete_experiment_tag
 get_experiment_by_id = experiment.get_experiment_by_name
 get_experiment_by_name = experiment.get_experiment_by_name
+create_experiment = experiment.create_experiment
 ModelMethodParams = parameters.ModelMethodParams
 ModelVersionChoice = parameters.ModelVersionChoice
 ModelForm = parameters.ModelForm

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/ML_management/sdk/experiment.py

@@ -160,3 +160,35 @@ def get_experiment_by_id(experiment_id: int) -> Experiment:
 
     experiment = send_graphql_request(op=op)
     return experiment["experimentFromId"]
+
+
+def create_experiment(
+    experiment_name: str, experiment_description: str = "", visibility: VisibilityOptions = VisibilityOptions.PRIVATE
+) -> Experiment:
+    """
+    Create experiment with given name and description.
+
+    Parameters
+    ----------
+    experiment_name: str
+        Name of the experiment.
+    experiment_description: str = ""
+        Description of the experiment. Default: "".
+    visibility: VisibilityOptions
+        Visibility of experiment.  Default: PRIVATE.
+
+    Returns
+    -------
+    Experiment
+        Instance of an experiment with meta information.
+    """
+    op = Operation(schema.Mutation)
+    create_experiment = op.create_experiment(
+        experiment_name=experiment_name, experiment_description=experiment_description, visibility=visibility.name
+    )
+    create_experiment.name()
+    create_experiment.description()
+
+    experiment = send_graphql_request(op=op, json_response=False)
+
+    return experiment.create_experiment

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/ML_management/sdk/job.py

@@ -16,6 +16,7 @@ from ML_management.graphql import schema
 from ML_management.graphql.schema import ExecutionJob, Param
 from ML_management.graphql.send_graphql_request import send_graphql_request
 from ML_management.mlmanagement import get_server_url
+from ML_management.mlmanagement.log_api import _raise_error
 from ML_management.mlmanagement.visibility_options import VisibilityOptions
 from ML_management.sdk.model import get_latest_model_version
 from ML_management.sdk.parameters import AnyDatasetLoaderForm, AnyModelForm, DatasetLoaderForm, ModelForm
@@ -314,6 +315,7 @@ def _get_logs(job_type: JobType, params: Dict[str, Any], stream: bool = True, fi
         return
 
     with AuthSession().get(url, stream=True) as resp:
+        _raise_error(resp)
         if file_name:
             with open(file_name, "a") as f:
                 for line in resp.iter_lines():

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ml-management
-Version: 0.6.3rc11
+Version: 0.6.3rc12
 Summary: Python implementation of model pattern, dataset
 Author-Email: ISPRAS MODIS <modis@ispras.ru>
 Maintainer: Maxim Ryndin

{ml_management-0.6.3rc11 → ml_management-0.6.3rc12}/pyproject.toml

@@ -26,7 +26,7 @@ readme = "README.md"
 maintainers = [
     { name = "Maxim Ryndin" },
 ]
-version = "0.6.3rc11"
+version = "0.6.3rc12"
 
 [build-system]
 requires = [
@@ -35,5 +35,5 @@ requires = [
 build-backend = "pdm.backend"
 
 [tool.pdm.version]
-source = "call"
-getter = "ML_management.version:get_version"
+source = "file"
+path = "ML_management/__init__.py"

ml_management-0.6.3rc11/ML_management/__init__.py

@@ -1,3 +0,0 @@
-from ML_management.version import get_version
-
-__version__ = get_version()

ml_management-0.6.3rc11/ML_management/version.py

@@ -1,6 +0,0 @@
-def get_version():
-    try:
-        with open("./VERSION", encoding="utf-8") as f:
-            return f.read()
-    except Exception:
-        return "unknown"