ml-management 0.5.1rc1__tar.gz → 0.5.1rc2__tar.gz

Sign up to get free protection for your applications and to get access to all the features.
Files changed (104) hide show
  1. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/load_api.py +15 -3
  2. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/PKG-INFO +1 -1
  3. ml-management-0.5.1rc2/VERSION +1 -0
  4. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ml_management.egg-info/PKG-INFO +1 -1
  5. ml-management-0.5.1rc1/VERSION +0 -1
  6. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/MANIFEST.in +0 -0
  7. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/__init__.py +0 -0
  8. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/__init__.py +0 -0
  9. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/collector_pattern.py +0 -0
  10. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/collector_pattern_to_methods_map.py +0 -0
  11. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/collectors.py +0 -0
  12. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/dummy/__init__.py +0 -0
  13. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/dummy/dummy_collector.py +0 -0
  14. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/s3/__init__.py +0 -0
  15. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/s3/s3collector.py +0 -0
  16. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/topic_markers/__init__.py +0 -0
  17. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/topic_markers/api_schema.py +0 -0
  18. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/collectors/topic_markers/topic_markers_collector.py +0 -0
  19. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/__init__.py +0 -0
  20. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/base_splits_dataset_loader.py +0 -0
  21. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/dataset_loader_pattern.py +0 -0
  22. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/dataset_loader_pattern_to_methods_map.py +0 -0
  23. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/poisoned_images_dataset_loader.py +0 -0
  24. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/templates/__init__.py +0 -0
  25. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/templates/dummy_dataset_loader/__init__.py +0 -0
  26. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/templates/dummy_dataset_loader/conda.yaml +0 -0
  27. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/templates/dummy_dataset_loader/dummy_dataset.py +0 -0
  28. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/dataset_loader/templates/upload.py +0 -0
  29. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/__init__.py +0 -0
  30. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/base_executor.py +0 -0
  31. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/executor_pattern.py +0 -0
  32. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/executor_pattern_to_methods_map.py +0 -0
  33. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/no_model_executor_pattern.py +0 -0
  34. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/patterns.py +0 -0
  35. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/__init__.py +0 -0
  36. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/eval/__init__.py +0 -0
  37. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/eval/conda.yaml +0 -0
  38. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/eval/eval_executor.py +0 -0
  39. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/finetune/__init__.py +0 -0
  40. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/finetune/conda.yaml +0 -0
  41. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/finetune/finetune_executor.py +0 -0
  42. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/train/__init__.py +0 -0
  43. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/train/conda.yaml +0 -0
  44. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/train/train_executor.py +0 -0
  45. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/templates/upload.py +0 -0
  46. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/executor/upload_model_mode.py +0 -0
  47. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/__init__.py +0 -0
  48. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/backend_api.py +0 -0
  49. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/base_exceptions.py +0 -0
  50. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/jsonschema_exceptions.py +0 -0
  51. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/jsonschema_inference.py +0 -0
  52. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/log_api.py +0 -0
  53. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/mlmanager.py +0 -0
  54. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/model_type.py +0 -0
  55. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/module_finder.py +0 -0
  56. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/server_mlmanager_exceptions.py +0 -0
  57. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/session.py +0 -0
  58. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/singleton_pattern.py +0 -0
  59. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/utils.py +0 -0
  60. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/variables.py +0 -0
  61. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/visibility_options.py +0 -0
  62. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/__init__.py +0 -0
  63. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/model_type_to_methods_map.py +0 -0
  64. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/__init__.py +0 -0
  65. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/evaluatable_model.py +0 -0
  66. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/gradient_model.py +0 -0
  67. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/model_pattern.py +0 -0
  68. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/model_with_losses.py +0 -0
  69. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/model_with_metrics.py +0 -0
  70. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/preprocessor.py +0 -0
  71. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/retrainable_model.py +0 -0
  72. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/rich_python_model.py +0 -0
  73. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/target_layer.py +0 -0
  74. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/torch_model.py +0 -0
  75. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/trainable_model.py +0 -0
  76. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/model/patterns/transformer.py +0 -0
  77. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/registry/__init__.py +0 -0
  78. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/registry/exceptions.py +0 -0
  79. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/s3/__init__.py +0 -0
  80. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/s3/manager.py +0 -0
  81. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/s3/utils.py +0 -0
  82. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/__init__.py +0 -0
  83. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/dataset_loader.py +0 -0
  84. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/executor.py +0 -0
  85. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/experiment.py +0 -0
  86. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/job.py +0 -0
  87. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/model.py +0 -0
  88. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/parameters.py +0 -0
  89. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/schema.py +0 -0
  90. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/sdk/sdk.py +0 -0
  91. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/test_sdk/__init__.py +0 -0
  92. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/test_sdk/test_sdk.py +0 -0
  93. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/tests/__init__.py +0 -0
  94. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/tests/test_jsonschema_inference.py +0 -0
  95. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/uploader_data/__init__.py +0 -0
  96. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/uploader_data/s3_uploader.py +0 -0
  97. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/uploader_data/utils.py +0 -0
  98. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/README.md +0 -0
  99. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ml_management.egg-info/SOURCES.txt +0 -0
  100. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ml_management.egg-info/dependency_links.txt +0 -0
  101. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ml_management.egg-info/requires.txt +0 -0
  102. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ml_management.egg-info/top_level.txt +0 -0
  103. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/setup.cfg +0 -0
  104. {ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/setup.py +0 -0
{ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ML_management/mlmanagement/load_api.py RENAMED
@@ -17,6 +17,8 @@ from ML_management.mlmanagement.log_api import _raise_error
17
17
  from ML_management.mlmanagement.model_type import ModelType
18
18
  from ML_management.mlmanagement.session import AuthSession
19
19
  from ML_management.mlmanagement.variables import get_log_service_url
20
+ from mlflow.models.model import MLMODEL_FILE_NAME
21
+ from mlflow.pyfunc import DATA, FLAVOR_NAME
20
22
  from mlflow.pyfunc.model import CONFIG_KEY_ARTIFACTS
21
23
 
22
24
  MLCONFIG = "MLConfig.yaml"
@@ -210,10 +212,20 @@ def _load_model_src(local_path: str, kwargs_for_init: Optional[dict]):
210
212
  if not kwargs_for_init:
211
213
  kwargs_for_init = {}
212
214
  config_path = os.path.join(local_path, MLCONFIG)
213
- with open(config_path) as file:
214
- conf = yaml.safe_load(file)
215
+ if os.path.exists(config_path):
216
+ with open(config_path) as file:
217
+ conf = yaml.safe_load(file)
215
218
 
216
- load_model_path = os.path.join(local_path, conf["load_model_path"])
219
+ load_model_path = os.path.join(local_path, conf["load_model_path"])
220
+
221
+ elif os.path.exists(os.path.join(local_path, MLMODEL_FILE_NAME)):
222
+ with open(os.path.join(local_path, MLMODEL_FILE_NAME)) as file:
223
+ conf = yaml.safe_load(file)
224
+
225
+ load_model_path = os.path.join(local_path, conf["flavors"][FLAVOR_NAME][DATA])
226
+
227
+ else:
228
+ raise RuntimeError("MLConfig does not exist.")
217
229
 
218
230
  from ML_management.mlmanagement.utils import INIT_FUNCTION_NAME # circular import
219
231
 
{ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ml-management
3
- Version: 0.5.1rc1
3
+ Version: 0.5.1rc2
4
4
  Summary: Python implementation of model pattern, dataset
5
5
  Author: ISPRAS MODIS
6
6
  Author-email: modis@ispras.ru
ml-management-0.5.1rc2/VERSION ADDED
@@ -0,0 +1 @@
1
+ 0.5.1rc2
{ml-management-0.5.1rc1 → ml-management-0.5.1rc2}/ml_management.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ml-management
3
- Version: 0.5.1rc1
3
+ Version: 0.5.1rc2
4
4
  Summary: Python implementation of model pattern, dataset
5
5
  Author: ISPRAS MODIS
6
6
  Author-email: modis@ispras.ru
ml-management-0.5.1rc1/VERSION DELETED
@@ -1 +0,0 @@
1
- 0.5.1rc1