ml-management 0.6.3rc18__tar.gz → 0.6.3rc19__tar.gz

ml_management-0.6.3rc19/ML_management/__init__.py

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

{ml_management-0.6.3rc18 → ml_management-0.6.3rc19}/ML_management/graphql/schema.py

@@ -267,8 +267,10 @@ class ModelVersionChoice(sgqlc.types.Input):
 
 class ObjectFilterSettings(sgqlc.types.Input):
     __schema__ = schema
-    __field_names__ = ('name',)
-    name = sgqlc.types.Field(sgqlc.types.non_null(String), graphql_name='name')
+    __field_names__ = ('name', 'tag')
+    name = sgqlc.types.Field(String, graphql_name='name')
+
+    tag = sgqlc.types.Field('TagFilterSettings', graphql_name='tag')
 
 
 
@@ -283,8 +285,10 @@ class ObjectSortBySortingInput(sgqlc.types.Input):
 
 class ObjectVersionFilterSettings(sgqlc.types.Input):
     __schema__ = schema
-    __field_names__ = ('version',)
-    version = sgqlc.types.Field(sgqlc.types.non_null(Int), graphql_name='version')
+    __field_names__ = ('tag', 'version')
+    tag = sgqlc.types.Field('TagFilterSettings', graphql_name='tag')
+
+    version = sgqlc.types.Field(Int, graphql_name='version')
 
 
 
@@ -351,6 +355,15 @@ class RoleObjectVersionInput(sgqlc.types.Input):
 
 
 
+class TagFilterSettings(sgqlc.types.Input):
+    __schema__ = schema
+    __field_names__ = ('key', 'value')
+    key = sgqlc.types.Field(String, graphql_name='key')
+
+    value = sgqlc.types.Field(String, graphql_name='value')
+
+
+
 class TimestampInterval(sgqlc.types.Input):
     __schema__ = schema
     __field_names__ = ('end', 'start')
@@ -496,7 +509,7 @@ class DatasetLoaderInfo(sgqlc.types.Type):
       `null`)
     '''
 
-    tags = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     visibility = sgqlc.types.Field(sgqlc.types.non_null(VisibilityOptions), graphql_name='visibility')
 
@@ -549,7 +562,7 @@ class DatasetLoaderVersionInfo(sgqlc.types.Type):
     * `offset` (`Int`)None (default: `null`)
     '''
 
-    tags = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     version = sgqlc.types.Field(sgqlc.types.non_null(Int), graphql_name='version')
 
@@ -672,7 +685,7 @@ class ExecutorInfo(sgqlc.types.Type):
       `null`)
     '''
 
-    tags = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     visibility = sgqlc.types.Field(sgqlc.types.non_null(VisibilityOptions), graphql_name='visibility')
 
@@ -922,7 +935,7 @@ class ExecutorVersionInfo(sgqlc.types.Type):
     * `offset` (`Int`)None (default: `null`)
     '''
 
-    tags = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     upload_model_modes = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='uploadModelModes')
 
@@ -1014,7 +1027,7 @@ class Experiment(sgqlc.types.Type):
       `null`)
     '''
 
-    tags = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     visibility = sgqlc.types.Field(sgqlc.types.non_null(VisibilityOptions), graphql_name='visibility')
 
@@ -1266,7 +1279,7 @@ class ModelInfo(sgqlc.types.Type):
       `null`)
     '''
 
-    tags = sgqlc.types.Field(JSON, graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     visibility = sgqlc.types.Field(sgqlc.types.non_null(VisibilityOptions), graphql_name='visibility')
 
@@ -1574,7 +1587,7 @@ class ModelVersionInfo(sgqlc.types.Type):
 
     source_model_version = sgqlc.types.Field('ModelVersionInfo', graphql_name='sourceModelVersion')
 
-    tags = sgqlc.types.Field(sgqlc.types.non_null(JSON), graphql_name='tags')
+    tags = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null('Tag'))), graphql_name='tags')
 
     upload_model_type = sgqlc.types.Field(sgqlc.types.non_null(UploadModelType), graphql_name='uploadModelType')
 
@@ -1597,7 +1610,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_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_job', 'stop_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', 'reset_dataset_loader_tags', 'reset_dataset_loader_version_tags', 'reset_executor_tags', 'reset_executor_version_tags', 'reset_experiment_tags', 'reset_model_tags', 'reset_model_version_tags', 'serve_model', 'set_dataset_loader_tags', 'set_dataset_loader_version_tags', 'set_executor_tags', 'set_executor_version_tags', 'set_experiment_tags', 'set_model_tags', 'set_model_version_tags', 'start_job', 'stop_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)),
 ))
@@ -1687,12 +1700,14 @@ class Mutation(sgqlc.types.Type):
     delete_dataset_loader_tag = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderInfo), graphql_name='deleteDatasetLoaderTag', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `name` (`String!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     delete_dataset_loader_version_from_name_version = sgqlc.types.Field(Boolean, graphql_name='deleteDatasetLoaderVersionFromNameVersion', args=sgqlc.types.ArgDict((
@@ -1707,12 +1722,14 @@ class Mutation(sgqlc.types.Type):
     delete_dataset_loader_version_tag = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderVersionInfo), graphql_name='deleteDatasetLoaderVersionTag', args=sgqlc.types.ArgDict((
         ('dataset_loader_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='datasetLoaderVersion', default=None)),
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `dataset_loader_version` (`ObjectVersionInput!`)None
     * `key` (`String!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     delete_executor = sgqlc.types.Field(sgqlc.types.non_null(Boolean), graphql_name='deleteExecutor', args=sgqlc.types.ArgDict((
@@ -1727,12 +1744,14 @@ class Mutation(sgqlc.types.Type):
     delete_executor_tag = sgqlc.types.Field(sgqlc.types.non_null(ExecutorInfo), graphql_name='deleteExecutorTag', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `name` (`String!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     delete_executor_version_from_name_version = sgqlc.types.Field(Boolean, graphql_name='deleteExecutorVersionFromNameVersion', args=sgqlc.types.ArgDict((
@@ -1747,23 +1766,27 @@ class Mutation(sgqlc.types.Type):
     delete_executor_version_tag = sgqlc.types.Field(sgqlc.types.non_null(ExecutorVersionInfo), graphql_name='deleteExecutorVersionTag', args=sgqlc.types.ArgDict((
         ('executor_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='executorVersion', default=None)),
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `executor_version` (`ObjectVersionInput!`)None
     * `key` (`String!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     delete_experiment_tag = sgqlc.types.Field(sgqlc.types.non_null(Experiment), graphql_name='deleteExperimentTag', args=sgqlc.types.ArgDict((
         ('experiment_id', sgqlc.types.Arg(sgqlc.types.non_null(Int), graphql_name='experimentId', default=None)),
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `experiment_id` (`Int!`)None
     * `key` (`String!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     delete_model = sgqlc.types.Field(sgqlc.types.non_null(Boolean), graphql_name='deleteModel', args=sgqlc.types.ArgDict((
@@ -1778,12 +1801,14 @@ class Mutation(sgqlc.types.Type):
     delete_model_tag = sgqlc.types.Field(sgqlc.types.non_null(ModelInfo), graphql_name='deleteModelTag', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `name` (`String!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     delete_model_version_from_name_version = sgqlc.types.Field(sgqlc.types.non_null(Boolean), graphql_name='deleteModelVersionFromNameVersion', args=sgqlc.types.ArgDict((
@@ -1798,12 +1823,14 @@ class Mutation(sgqlc.types.Type):
     delete_model_version_tag = sgqlc.types.Field(sgqlc.types.non_null(ModelVersionInfo), graphql_name='deleteModelVersionTag', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('model_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='modelVersion', default=None)),
+        ('value', sgqlc.types.Arg(String, graphql_name='value', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `model_version` (`ObjectVersionInput!`)None
+    * `value` (`String`)None (default: `null`)
     '''
 
     log_metric = sgqlc.types.Field(sgqlc.types.non_null(Boolean), graphql_name='logMetric', args=sgqlc.types.ArgDict((
@@ -1879,6 +1906,111 @@ class Mutation(sgqlc.types.Type):
     * `new_name` (`String!`)None
     '''
 
+    reset_dataset_loader_tags = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderInfo), graphql_name='resetDatasetLoaderTags', args=sgqlc.types.ArgDict((
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `key` (`String!`)None
+    * `name` (`String!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
+    reset_dataset_loader_version_tags = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderVersionInfo), graphql_name='resetDatasetLoaderVersionTags', args=sgqlc.types.ArgDict((
+        ('dataset_loader_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='datasetLoaderVersion', default=None)),
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `dataset_loader_version` (`ObjectVersionInput!`)None
+    * `key` (`String!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
+    reset_executor_tags = sgqlc.types.Field(sgqlc.types.non_null(ExecutorInfo), graphql_name='resetExecutorTags', args=sgqlc.types.ArgDict((
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `key` (`String!`)None
+    * `name` (`String!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
+    reset_executor_version_tags = sgqlc.types.Field(sgqlc.types.non_null(ExecutorVersionInfo), graphql_name='resetExecutorVersionTags', args=sgqlc.types.ArgDict((
+        ('executor_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='executorVersion', default=None)),
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `executor_version` (`ObjectVersionInput!`)None
+    * `key` (`String!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
+    reset_experiment_tags = sgqlc.types.Field(sgqlc.types.non_null(Experiment), graphql_name='resetExperimentTags', args=sgqlc.types.ArgDict((
+        ('experiment_id', sgqlc.types.Arg(sgqlc.types.non_null(Int), graphql_name='experimentId', default=None)),
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `experiment_id` (`Int!`)None
+    * `key` (`String!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
+    reset_model_tags = sgqlc.types.Field(sgqlc.types.non_null(ModelInfo), graphql_name='resetModelTags', args=sgqlc.types.ArgDict((
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `key` (`String!`)None
+    * `name` (`String!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
+    reset_model_version_tags = sgqlc.types.Field(sgqlc.types.non_null(ModelVersionInfo), graphql_name='resetModelVersionTags', args=sgqlc.types.ArgDict((
+        ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
+        ('model_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='modelVersion', default=None)),
+        ('new_key', sgqlc.types.Arg(String, graphql_name='newKey', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
+))
+    )
+    '''Arguments:
+
+    * `key` (`String!`)None
+    * `model_version` (`ObjectVersionInput!`)None
+    * `new_key` (`String`)None (default: `null`)
+    * `values` (`[String!]!`)None
+    '''
+
     serve_model = sgqlc.types.Field(sgqlc.types.non_null(Boolean), graphql_name='serveModel', args=sgqlc.types.ArgDict((
         ('serving_parameters', sgqlc.types.Arg(sgqlc.types.non_null(ModelServingInput), graphql_name='servingParameters', default=None)),
 ))
@@ -1888,95 +2020,95 @@ class Mutation(sgqlc.types.Type):
     * `serving_parameters` (`ModelServingInput!`)None
     '''
 
-    set_dataset_loader_tag = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderInfo), graphql_name='setDatasetLoaderTag', args=sgqlc.types.ArgDict((
+    set_dataset_loader_tags = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderInfo), graphql_name='setDatasetLoaderTags', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `name` (`String!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
-    set_dataset_loader_version_tag = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderVersionInfo), graphql_name='setDatasetLoaderVersionTag', args=sgqlc.types.ArgDict((
+    set_dataset_loader_version_tags = sgqlc.types.Field(sgqlc.types.non_null(DatasetLoaderVersionInfo), graphql_name='setDatasetLoaderVersionTags', args=sgqlc.types.ArgDict((
         ('dataset_loader_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='datasetLoaderVersion', default=None)),
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `dataset_loader_version` (`ObjectVersionInput!`)None
     * `key` (`String!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
-    set_executor_tag = sgqlc.types.Field(sgqlc.types.non_null(ExecutorInfo), graphql_name='setExecutorTag', args=sgqlc.types.ArgDict((
+    set_executor_tags = sgqlc.types.Field(sgqlc.types.non_null(ExecutorInfo), graphql_name='setExecutorTags', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `name` (`String!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
-    set_executor_version_tag = sgqlc.types.Field(sgqlc.types.non_null(ExecutorVersionInfo), graphql_name='setExecutorVersionTag', args=sgqlc.types.ArgDict((
+    set_executor_version_tags = sgqlc.types.Field(sgqlc.types.non_null(ExecutorVersionInfo), graphql_name='setExecutorVersionTags', args=sgqlc.types.ArgDict((
         ('executor_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='executorVersion', default=None)),
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `executor_version` (`ObjectVersionInput!`)None
     * `key` (`String!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
-    set_experiment_tag = sgqlc.types.Field(sgqlc.types.non_null(Experiment), graphql_name='setExperimentTag', args=sgqlc.types.ArgDict((
+    set_experiment_tags = sgqlc.types.Field(sgqlc.types.non_null(Experiment), graphql_name='setExperimentTags', args=sgqlc.types.ArgDict((
         ('experiment_id', sgqlc.types.Arg(sgqlc.types.non_null(Int), graphql_name='experimentId', default=None)),
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `experiment_id` (`Int!`)None
     * `key` (`String!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
-    set_model_tag = sgqlc.types.Field(sgqlc.types.non_null(ModelInfo), graphql_name='setModelTag', args=sgqlc.types.ArgDict((
+    set_model_tags = sgqlc.types.Field(sgqlc.types.non_null(ModelInfo), graphql_name='setModelTags', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='name', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `name` (`String!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
-    set_model_version_tag = sgqlc.types.Field(sgqlc.types.non_null(ModelVersionInfo), graphql_name='setModelVersionTag', args=sgqlc.types.ArgDict((
+    set_model_version_tags = sgqlc.types.Field(sgqlc.types.non_null(ModelVersionInfo), graphql_name='setModelVersionTags', args=sgqlc.types.ArgDict((
         ('key', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='key', default=None)),
         ('model_version', sgqlc.types.Arg(sgqlc.types.non_null(ObjectVersionInput), graphql_name='modelVersion', default=None)),
-        ('value', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='value', default=None)),
+        ('values', sgqlc.types.Arg(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values', default=None)),
 ))
     )
     '''Arguments:
 
     * `key` (`String!`)None
     * `model_version` (`ObjectVersionInput!`)None
-    * `value` (`String!`)None
+    * `values` (`[String!]!`)None
     '''
 
     start_job = sgqlc.types.Field(sgqlc.types.non_null(ExecutionJob), graphql_name='startJob', args=sgqlc.types.ArgDict((
@@ -2523,6 +2655,15 @@ class Subscription(sgqlc.types.Type):
 
 
 
+class Tag(sgqlc.types.Type):
+    __schema__ = schema
+    __field_names__ = ('key', 'values')
+    key = sgqlc.types.Field(sgqlc.types.non_null(String), graphql_name='key')
+
+    values = sgqlc.types.Field(sgqlc.types.non_null(sgqlc.types.list_of(sgqlc.types.non_null(String))), graphql_name='values')
+
+
+
 class User(sgqlc.types.Type):
     __schema__ = schema
     __field_names__ = ('id',)

{ml_management-0.6.3rc18 → ml_management-0.6.3rc19}/ML_management/graphql/send_graphql_request.py

@@ -9,7 +9,7 @@ from ML_management.session import AuthSession
 from ML_management.variables import get_server_url
 
 
-def send_graphql_request(op: Operation, json_response: bool = True, timeout=None) -> Any:
+def send_graphql_request(op: Operation, json_response: bool = True, timeout=60) -> Any:
     """Send request to server and process the response."""
     json_data = AuthSession().sgqlc_request(op, timeout)
 

{ml_management-0.6.3rc18 → ml_management-0.6.3rc19}/ML_management/mlmanagement/active_job.py

@@ -39,7 +39,7 @@ class ActiveJob:
         status = "SUCCESSFUL"
         if exc_type:
             exception_traceback = traceback.format_exc()
-            message = str(exc_val)
+            message = ": ".join([exc_type.__name__, str(exc_val)])
             status = "FAILED"
 
         op = Operation(schema.Mutation)

{ml_management-0.6.3rc18 → ml_management-0.6.3rc19}/ML_management/mlmanagement/batcher.py

@@ -11,7 +11,11 @@ from ML_management.graphql import schema
 from ML_management.graphql.schema import MetricInput
 from ML_management.graphql.send_graphql_request import send_graphql_request
 from ML_management.singleton_pattern import Singleton
-from ML_management.variables import METRIC_ACCUMULATION_DURATION, TIMEOUT_LOG_METRIC_BATCH
+from ML_management.variables import (
+    get_butch_polling_frequency,
+    get_metric_accumulation_duration,
+    get_timeout_log_metric_batch,
+)
 
 
 class Batcher(metaclass=Singleton):
@@ -38,12 +42,13 @@ class Batcher(metaclass=Singleton):
         self.batch = deque()
         if self.thread:
             self.stop_event.set()
-            self.thread.join()
+            self.stop_event = Event()
             self.thread = None
 
     def log_batch_metric(self, stop_event):
+        metric_accumulation_duration = get_metric_accumulation_duration()
         while not stop_event.is_set():
-            time.sleep(METRIC_ACCUMULATION_DURATION)
+            time.sleep(metric_accumulation_duration)
             if len(self.batch) == 0:
                 continue
             metrics_to_log = self.batch.copy()
@@ -56,7 +61,7 @@ class Batcher(metaclass=Singleton):
                 secret_uuid=variables.secret_uuid,
             )
             try:
-                send_graphql_request(op, json_response=False, timeout=5)
+                send_graphql_request(op, json_response=False)
                 for _ in range(len(metrics_to_log)):
                     self.batch.popleft()
             except Exception as err:
@@ -64,6 +69,7 @@ class Batcher(metaclass=Singleton):
 
     def wait_log_metrics(self):
         stop_time = time.time()
-        while self.batch and time.time() - stop_time < TIMEOUT_LOG_METRIC_BATCH:
-            time.sleep(1)
+        butch_polling_frequency = get_butch_polling_frequency()
+        while self.batch and time.time() - stop_time < get_timeout_log_metric_batch():
+            time.sleep(butch_polling_frequency)
         self._stop_batching()

{ml_management-0.6.3rc18 → ml_management-0.6.3rc19}/ML_management/mlmanagement/log_api.py

@@ -52,7 +52,7 @@ def _log_object_src(
     description: str,
     experiment_name: str,
     model_type: ModelType = ModelType.MODEL,
-    model_version_tags: Optional[Dict[str, str]] = None,
+    model_version_tags: Optional[Dict[str, Union[str, list]]] = None,
     registered_model_name: str = "default_name",
     source_model_name=None,
     source_model_version=None,
@@ -113,6 +113,12 @@ def _log_object_src(
     del UploadModelMode  # need to delete this because it is not JSON serializable
     if create_venv_pack:
         validate_predict_config(path=os.path.join(f"{model_path}", "artifacts", f"{FILENAME_FOR_INFERENCE_CONFIG}"))
+
+    if model_version_tags:
+        for key, value in model_version_tags.items():
+            if not isinstance(value, list):
+                model_version_tags[key] = [value]
+
     old_python_path = sys.path.copy()
     old_sys_modules = sys.modules.copy()
     try:
@@ -277,7 +283,7 @@ def log_model_src(
     registered_name: str,
     description: str,
     start_build: bool = True,
-    model_version_tags: Optional[Dict[str, str]] = None,
+    model_version_tags: Optional[Dict[str, Union[str, list]]] = None,
     experiment_name: str = DEFAULT_EXPERIMENT,
     create_venv_pack: bool = False,
     visibility: VisibilityOptions = VisibilityOptions.PRIVATE,
@@ -301,7 +307,7 @@ def log_model_src(
     start_build: bool = True
         Whether to start building the image with the model requirements immediately after uploading.
         This can speed up the subsequent build of the task image. Default: True.
-    model_version_tags: Optional[Dict[str, str]] = None
+    model_version_tags: Optional[Dict[str, Union[str, list]]] = None
         Define model version tags. Default: None.
     create_venv_pack: bool = False
         Whether to prepare venv pack for future inference. Default: False.

{ml_management-0.6.3rc18 → ml_management-0.6.3rc19}/ML_management/sdk/__init__.py

@@ -35,10 +35,11 @@ cancel_job = job.cancel_job
 cancel_build_job_for_model_version = model.cancel_build_job_for_model_version
 cancel_build_job_for_executor_version = executor.cancel_build_job_for_executor_version
 cancel_venv_build_job_for_model_version = model.cancel_venv_build_job_for_model_version
-set_experiment_tag = experiment.set_experiment_tag
+set_experiment_tags = experiment.set_experiment_tags
+reset_experiment_tags = experiment.reset_experiment_tags
 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_id = experiment.get_experiment_by_id
 get_experiment_by_name = experiment.get_experiment_by_name
 create_experiment = experiment.create_experiment
 ModelMethodParams = parameters.ModelMethodParams
@@ -61,9 +62,12 @@ get_dataset_loader_version = dataset_loader.get_dataset_loader_version
 delete_dataset_loader_version_tag = dataset_loader.delete_dataset_loader_version_tag
 delete_executor_version_tag = executor.delete_executor_version_tag
 delete_model_version_tag = model.delete_model_version_tag
-set_dataset_loader_version_tag = dataset_loader.set_dataset_loader_version_tag
-set_executor_version_tag = executor.set_executor_version_tag
-set_model_version_tag = model.set_model_version_tag
+set_dataset_loader_version_tags = dataset_loader.set_dataset_loader_version_tags
+reset_dataset_loader_version_tags = dataset_loader.reset_dataset_loader_version_tags
+set_executor_version_tags = executor.set_executor_version_tags
+reset_executor_version_tags = executor.reset_executor_version_tags
+set_model_version_tags = model.set_model_version_tags
+reset_model_version_tags = model.reset_model_version_tags
 set_dataset_loader_version_description = dataset_loader.set_dataset_loader_version_description
 set_executor_version_description = executor.set_executor_version_description
 set_model_version_description = model.set_model_version_description
@@ -73,9 +77,12 @@ set_dataset_loader_description = dataset_loader.set_dataset_loader_description
 delete_model_tag = model.delete_model_tag
 delete_executor_tag = executor.delete_executor_tag
 delete_dataset_loader_tag = dataset_loader.delete_dataset_loader_tag
-set_dataset_loader_tag = dataset_loader.set_dataset_loader_tag
-set_executor_tag = executor.set_executor_tag
-set_model_tag = model.set_model_tag
+set_dataset_loader_tags = dataset_loader.set_dataset_loader_tags
+reset_dataset_loader_tags = dataset_loader.reset_dataset_loader_tags
+set_executor_tags = executor.set_executor_tags
+reset_executor_tags = executor.reset_executor_tags
+set_model_tags = model.set_model_tags
+reset_model_tags = model.reset_model_tags
 get_latest_model_version = model.get_latest_model_version
 get_latest_executor_version = executor.get_latest_executor_version
 get_latest_dataset_loader_version = dataset_loader.get_latest_dataset_loader_version
@@ -91,3 +98,10 @@ set_executor_visibility = executor.set_executor_visibility
 set_dataset_loader_visibility = dataset_loader.set_dataset_loader_visibility
 set_experiment_visibility = experiment.set_experiment_visibility
 set_job_visibility = job.set_job_visibility
+
+pagination_model = model.pagination_model
+pagination_model_version = model.pagination_model_version
+pagination_executor = executor.pagination_executor
+pagination_executor_version = executor.pagination_executor_version
+pagination_dataset_loader = dataset_loader.pagination_dataset_loader
+pagination_dataset_loader_version = dataset_loader.pagination_dataset_loader_version