genesis-flow 1.0.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (645) hide show
  1. genesis_flow-1.0.0.dist-info/METADATA +822 -0
  2. genesis_flow-1.0.0.dist-info/RECORD +645 -0
  3. genesis_flow-1.0.0.dist-info/WHEEL +5 -0
  4. genesis_flow-1.0.0.dist-info/entry_points.txt +19 -0
  5. genesis_flow-1.0.0.dist-info/licenses/LICENSE.txt +202 -0
  6. genesis_flow-1.0.0.dist-info/top_level.txt +1 -0
  7. mlflow/__init__.py +367 -0
  8. mlflow/__main__.py +3 -0
  9. mlflow/ag2/__init__.py +56 -0
  10. mlflow/ag2/ag2_logger.py +294 -0
  11. mlflow/anthropic/__init__.py +40 -0
  12. mlflow/anthropic/autolog.py +129 -0
  13. mlflow/anthropic/chat.py +144 -0
  14. mlflow/artifacts/__init__.py +268 -0
  15. mlflow/autogen/__init__.py +144 -0
  16. mlflow/autogen/chat.py +142 -0
  17. mlflow/azure/__init__.py +26 -0
  18. mlflow/azure/auth_handler.py +257 -0
  19. mlflow/azure/client.py +319 -0
  20. mlflow/azure/config.py +120 -0
  21. mlflow/azure/connection_factory.py +340 -0
  22. mlflow/azure/exceptions.py +27 -0
  23. mlflow/azure/stores.py +327 -0
  24. mlflow/azure/utils.py +183 -0
  25. mlflow/bedrock/__init__.py +45 -0
  26. mlflow/bedrock/_autolog.py +202 -0
  27. mlflow/bedrock/chat.py +122 -0
  28. mlflow/bedrock/stream.py +160 -0
  29. mlflow/bedrock/utils.py +43 -0
  30. mlflow/cli.py +707 -0
  31. mlflow/client.py +12 -0
  32. mlflow/config/__init__.py +56 -0
  33. mlflow/crewai/__init__.py +79 -0
  34. mlflow/crewai/autolog.py +253 -0
  35. mlflow/crewai/chat.py +29 -0
  36. mlflow/data/__init__.py +75 -0
  37. mlflow/data/artifact_dataset_sources.py +170 -0
  38. mlflow/data/code_dataset_source.py +40 -0
  39. mlflow/data/dataset.py +123 -0
  40. mlflow/data/dataset_registry.py +168 -0
  41. mlflow/data/dataset_source.py +110 -0
  42. mlflow/data/dataset_source_registry.py +219 -0
  43. mlflow/data/delta_dataset_source.py +167 -0
  44. mlflow/data/digest_utils.py +108 -0
  45. mlflow/data/evaluation_dataset.py +562 -0
  46. mlflow/data/filesystem_dataset_source.py +81 -0
  47. mlflow/data/http_dataset_source.py +145 -0
  48. mlflow/data/huggingface_dataset.py +258 -0
  49. mlflow/data/huggingface_dataset_source.py +118 -0
  50. mlflow/data/meta_dataset.py +104 -0
  51. mlflow/data/numpy_dataset.py +223 -0
  52. mlflow/data/pandas_dataset.py +231 -0
  53. mlflow/data/polars_dataset.py +352 -0
  54. mlflow/data/pyfunc_dataset_mixin.py +31 -0
  55. mlflow/data/schema.py +76 -0
  56. mlflow/data/sources.py +1 -0
  57. mlflow/data/spark_dataset.py +406 -0
  58. mlflow/data/spark_dataset_source.py +74 -0
  59. mlflow/data/spark_delta_utils.py +118 -0
  60. mlflow/data/tensorflow_dataset.py +350 -0
  61. mlflow/data/uc_volume_dataset_source.py +81 -0
  62. mlflow/db.py +27 -0
  63. mlflow/dspy/__init__.py +17 -0
  64. mlflow/dspy/autolog.py +197 -0
  65. mlflow/dspy/callback.py +398 -0
  66. mlflow/dspy/constant.py +1 -0
  67. mlflow/dspy/load.py +93 -0
  68. mlflow/dspy/save.py +393 -0
  69. mlflow/dspy/util.py +109 -0
  70. mlflow/dspy/wrapper.py +226 -0
  71. mlflow/entities/__init__.py +104 -0
  72. mlflow/entities/_mlflow_object.py +52 -0
  73. mlflow/entities/assessment.py +545 -0
  74. mlflow/entities/assessment_error.py +80 -0
  75. mlflow/entities/assessment_source.py +141 -0
  76. mlflow/entities/dataset.py +92 -0
  77. mlflow/entities/dataset_input.py +51 -0
  78. mlflow/entities/dataset_summary.py +62 -0
  79. mlflow/entities/document.py +48 -0
  80. mlflow/entities/experiment.py +109 -0
  81. mlflow/entities/experiment_tag.py +35 -0
  82. mlflow/entities/file_info.py +45 -0
  83. mlflow/entities/input_tag.py +35 -0
  84. mlflow/entities/lifecycle_stage.py +35 -0
  85. mlflow/entities/logged_model.py +228 -0
  86. mlflow/entities/logged_model_input.py +26 -0
  87. mlflow/entities/logged_model_output.py +32 -0
  88. mlflow/entities/logged_model_parameter.py +46 -0
  89. mlflow/entities/logged_model_status.py +74 -0
  90. mlflow/entities/logged_model_tag.py +33 -0
  91. mlflow/entities/metric.py +200 -0
  92. mlflow/entities/model_registry/__init__.py +29 -0
  93. mlflow/entities/model_registry/_model_registry_entity.py +13 -0
  94. mlflow/entities/model_registry/model_version.py +243 -0
  95. mlflow/entities/model_registry/model_version_deployment_job_run_state.py +44 -0
  96. mlflow/entities/model_registry/model_version_deployment_job_state.py +70 -0
  97. mlflow/entities/model_registry/model_version_search.py +25 -0
  98. mlflow/entities/model_registry/model_version_stages.py +25 -0
  99. mlflow/entities/model_registry/model_version_status.py +35 -0
  100. mlflow/entities/model_registry/model_version_tag.py +35 -0
  101. mlflow/entities/model_registry/prompt.py +73 -0
  102. mlflow/entities/model_registry/prompt_version.py +244 -0
  103. mlflow/entities/model_registry/registered_model.py +175 -0
  104. mlflow/entities/model_registry/registered_model_alias.py +35 -0
  105. mlflow/entities/model_registry/registered_model_deployment_job_state.py +39 -0
  106. mlflow/entities/model_registry/registered_model_search.py +25 -0
  107. mlflow/entities/model_registry/registered_model_tag.py +35 -0
  108. mlflow/entities/multipart_upload.py +74 -0
  109. mlflow/entities/param.py +49 -0
  110. mlflow/entities/run.py +97 -0
  111. mlflow/entities/run_data.py +84 -0
  112. mlflow/entities/run_info.py +188 -0
  113. mlflow/entities/run_inputs.py +59 -0
  114. mlflow/entities/run_outputs.py +43 -0
  115. mlflow/entities/run_status.py +41 -0
  116. mlflow/entities/run_tag.py +36 -0
  117. mlflow/entities/source_type.py +31 -0
  118. mlflow/entities/span.py +774 -0
  119. mlflow/entities/span_event.py +96 -0
  120. mlflow/entities/span_status.py +102 -0
  121. mlflow/entities/trace.py +317 -0
  122. mlflow/entities/trace_data.py +71 -0
  123. mlflow/entities/trace_info.py +220 -0
  124. mlflow/entities/trace_info_v2.py +162 -0
  125. mlflow/entities/trace_location.py +173 -0
  126. mlflow/entities/trace_state.py +39 -0
  127. mlflow/entities/trace_status.py +68 -0
  128. mlflow/entities/view_type.py +51 -0
  129. mlflow/environment_variables.py +866 -0
  130. mlflow/evaluation/__init__.py +16 -0
  131. mlflow/evaluation/assessment.py +369 -0
  132. mlflow/evaluation/evaluation.py +411 -0
  133. mlflow/evaluation/evaluation_tag.py +61 -0
  134. mlflow/evaluation/fluent.py +48 -0
  135. mlflow/evaluation/utils.py +201 -0
  136. mlflow/exceptions.py +213 -0
  137. mlflow/experiments.py +140 -0
  138. mlflow/gemini/__init__.py +81 -0
  139. mlflow/gemini/autolog.py +186 -0
  140. mlflow/gemini/chat.py +261 -0
  141. mlflow/genai/__init__.py +71 -0
  142. mlflow/genai/datasets/__init__.py +67 -0
  143. mlflow/genai/datasets/evaluation_dataset.py +131 -0
  144. mlflow/genai/evaluation/__init__.py +3 -0
  145. mlflow/genai/evaluation/base.py +411 -0
  146. mlflow/genai/evaluation/constant.py +23 -0
  147. mlflow/genai/evaluation/utils.py +244 -0
  148. mlflow/genai/judges/__init__.py +21 -0
  149. mlflow/genai/judges/databricks.py +404 -0
  150. mlflow/genai/label_schemas/__init__.py +153 -0
  151. mlflow/genai/label_schemas/label_schemas.py +209 -0
  152. mlflow/genai/labeling/__init__.py +159 -0
  153. mlflow/genai/labeling/labeling.py +250 -0
  154. mlflow/genai/optimize/__init__.py +13 -0
  155. mlflow/genai/optimize/base.py +198 -0
  156. mlflow/genai/optimize/optimizers/__init__.py +4 -0
  157. mlflow/genai/optimize/optimizers/base_optimizer.py +38 -0
  158. mlflow/genai/optimize/optimizers/dspy_mipro_optimizer.py +221 -0
  159. mlflow/genai/optimize/optimizers/dspy_optimizer.py +91 -0
  160. mlflow/genai/optimize/optimizers/utils/dspy_mipro_callback.py +76 -0
  161. mlflow/genai/optimize/optimizers/utils/dspy_mipro_utils.py +18 -0
  162. mlflow/genai/optimize/types.py +75 -0
  163. mlflow/genai/optimize/util.py +30 -0
  164. mlflow/genai/prompts/__init__.py +206 -0
  165. mlflow/genai/scheduled_scorers.py +431 -0
  166. mlflow/genai/scorers/__init__.py +26 -0
  167. mlflow/genai/scorers/base.py +492 -0
  168. mlflow/genai/scorers/builtin_scorers.py +765 -0
  169. mlflow/genai/scorers/scorer_utils.py +138 -0
  170. mlflow/genai/scorers/validation.py +165 -0
  171. mlflow/genai/utils/data_validation.py +146 -0
  172. mlflow/genai/utils/enum_utils.py +23 -0
  173. mlflow/genai/utils/trace_utils.py +211 -0
  174. mlflow/groq/__init__.py +42 -0
  175. mlflow/groq/_groq_autolog.py +74 -0
  176. mlflow/johnsnowlabs/__init__.py +888 -0
  177. mlflow/langchain/__init__.py +24 -0
  178. mlflow/langchain/api_request_parallel_processor.py +330 -0
  179. mlflow/langchain/autolog.py +147 -0
  180. mlflow/langchain/chat_agent_langgraph.py +340 -0
  181. mlflow/langchain/constant.py +1 -0
  182. mlflow/langchain/constants.py +1 -0
  183. mlflow/langchain/databricks_dependencies.py +444 -0
  184. mlflow/langchain/langchain_tracer.py +597 -0
  185. mlflow/langchain/model.py +919 -0
  186. mlflow/langchain/output_parsers.py +142 -0
  187. mlflow/langchain/retriever_chain.py +153 -0
  188. mlflow/langchain/runnables.py +527 -0
  189. mlflow/langchain/utils/chat.py +402 -0
  190. mlflow/langchain/utils/logging.py +671 -0
  191. mlflow/langchain/utils/serialization.py +36 -0
  192. mlflow/legacy_databricks_cli/__init__.py +0 -0
  193. mlflow/legacy_databricks_cli/configure/__init__.py +0 -0
  194. mlflow/legacy_databricks_cli/configure/provider.py +482 -0
  195. mlflow/litellm/__init__.py +175 -0
  196. mlflow/llama_index/__init__.py +22 -0
  197. mlflow/llama_index/autolog.py +55 -0
  198. mlflow/llama_index/chat.py +43 -0
  199. mlflow/llama_index/constant.py +1 -0
  200. mlflow/llama_index/model.py +577 -0
  201. mlflow/llama_index/pyfunc_wrapper.py +332 -0
  202. mlflow/llama_index/serialize_objects.py +188 -0
  203. mlflow/llama_index/tracer.py +561 -0
  204. mlflow/metrics/__init__.py +479 -0
  205. mlflow/metrics/base.py +39 -0
  206. mlflow/metrics/genai/__init__.py +25 -0
  207. mlflow/metrics/genai/base.py +101 -0
  208. mlflow/metrics/genai/genai_metric.py +771 -0
  209. mlflow/metrics/genai/metric_definitions.py +450 -0
  210. mlflow/metrics/genai/model_utils.py +371 -0
  211. mlflow/metrics/genai/prompt_template.py +68 -0
  212. mlflow/metrics/genai/prompts/__init__.py +0 -0
  213. mlflow/metrics/genai/prompts/v1.py +422 -0
  214. mlflow/metrics/genai/utils.py +6 -0
  215. mlflow/metrics/metric_definitions.py +619 -0
  216. mlflow/mismatch.py +34 -0
  217. mlflow/mistral/__init__.py +34 -0
  218. mlflow/mistral/autolog.py +71 -0
  219. mlflow/mistral/chat.py +135 -0
  220. mlflow/ml_package_versions.py +452 -0
  221. mlflow/models/__init__.py +97 -0
  222. mlflow/models/auth_policy.py +83 -0
  223. mlflow/models/cli.py +354 -0
  224. mlflow/models/container/__init__.py +294 -0
  225. mlflow/models/container/scoring_server/__init__.py +0 -0
  226. mlflow/models/container/scoring_server/nginx.conf +39 -0
  227. mlflow/models/dependencies_schemas.py +287 -0
  228. mlflow/models/display_utils.py +158 -0
  229. mlflow/models/docker_utils.py +211 -0
  230. mlflow/models/evaluation/__init__.py +23 -0
  231. mlflow/models/evaluation/_shap_patch.py +64 -0
  232. mlflow/models/evaluation/artifacts.py +194 -0
  233. mlflow/models/evaluation/base.py +1811 -0
  234. mlflow/models/evaluation/calibration_curve.py +109 -0
  235. mlflow/models/evaluation/default_evaluator.py +996 -0
  236. mlflow/models/evaluation/deprecated.py +23 -0
  237. mlflow/models/evaluation/evaluator_registry.py +80 -0
  238. mlflow/models/evaluation/evaluators/classifier.py +704 -0
  239. mlflow/models/evaluation/evaluators/default.py +233 -0
  240. mlflow/models/evaluation/evaluators/regressor.py +96 -0
  241. mlflow/models/evaluation/evaluators/shap.py +296 -0
  242. mlflow/models/evaluation/lift_curve.py +178 -0
  243. mlflow/models/evaluation/utils/metric.py +123 -0
  244. mlflow/models/evaluation/utils/trace.py +179 -0
  245. mlflow/models/evaluation/validation.py +434 -0
  246. mlflow/models/flavor_backend.py +93 -0
  247. mlflow/models/flavor_backend_registry.py +53 -0
  248. mlflow/models/model.py +1639 -0
  249. mlflow/models/model_config.py +150 -0
  250. mlflow/models/notebook_resources/agent_evaluation_template.html +235 -0
  251. mlflow/models/notebook_resources/eval_with_dataset_example.py +22 -0
  252. mlflow/models/notebook_resources/eval_with_synthetic_example.py +22 -0
  253. mlflow/models/python_api.py +369 -0
  254. mlflow/models/rag_signatures.py +128 -0
  255. mlflow/models/resources.py +321 -0
  256. mlflow/models/signature.py +662 -0
  257. mlflow/models/utils.py +2054 -0
  258. mlflow/models/wheeled_model.py +280 -0
  259. mlflow/openai/__init__.py +57 -0
  260. mlflow/openai/_agent_tracer.py +364 -0
  261. mlflow/openai/api_request_parallel_processor.py +131 -0
  262. mlflow/openai/autolog.py +509 -0
  263. mlflow/openai/constant.py +1 -0
  264. mlflow/openai/model.py +824 -0
  265. mlflow/openai/utils/chat_schema.py +367 -0
  266. mlflow/optuna/__init__.py +3 -0
  267. mlflow/optuna/storage.py +646 -0
  268. mlflow/plugins/__init__.py +72 -0
  269. mlflow/plugins/base.py +358 -0
  270. mlflow/plugins/builtin/__init__.py +24 -0
  271. mlflow/plugins/builtin/pytorch_plugin.py +150 -0
  272. mlflow/plugins/builtin/sklearn_plugin.py +158 -0
  273. mlflow/plugins/builtin/transformers_plugin.py +187 -0
  274. mlflow/plugins/cli.py +321 -0
  275. mlflow/plugins/discovery.py +340 -0
  276. mlflow/plugins/manager.py +465 -0
  277. mlflow/plugins/registry.py +316 -0
  278. mlflow/plugins/templates/framework_plugin_template.py +329 -0
  279. mlflow/prompt/constants.py +20 -0
  280. mlflow/prompt/promptlab_model.py +197 -0
  281. mlflow/prompt/registry_utils.py +248 -0
  282. mlflow/promptflow/__init__.py +495 -0
  283. mlflow/protos/__init__.py +0 -0
  284. mlflow/protos/assessments_pb2.py +174 -0
  285. mlflow/protos/databricks_artifacts_pb2.py +489 -0
  286. mlflow/protos/databricks_filesystem_service_pb2.py +196 -0
  287. mlflow/protos/databricks_managed_catalog_messages_pb2.py +95 -0
  288. mlflow/protos/databricks_managed_catalog_service_pb2.py +86 -0
  289. mlflow/protos/databricks_pb2.py +267 -0
  290. mlflow/protos/databricks_trace_server_pb2.py +374 -0
  291. mlflow/protos/databricks_uc_registry_messages_pb2.py +1249 -0
  292. mlflow/protos/databricks_uc_registry_service_pb2.py +170 -0
  293. mlflow/protos/facet_feature_statistics_pb2.py +296 -0
  294. mlflow/protos/internal_pb2.py +77 -0
  295. mlflow/protos/mlflow_artifacts_pb2.py +336 -0
  296. mlflow/protos/model_registry_pb2.py +1073 -0
  297. mlflow/protos/scalapb/__init__.py +0 -0
  298. mlflow/protos/scalapb/scalapb_pb2.py +104 -0
  299. mlflow/protos/service_pb2.py +2600 -0
  300. mlflow/protos/unity_catalog_oss_messages_pb2.py +457 -0
  301. mlflow/protos/unity_catalog_oss_service_pb2.py +130 -0
  302. mlflow/protos/unity_catalog_prompt_messages_pb2.py +447 -0
  303. mlflow/protos/unity_catalog_prompt_messages_pb2_grpc.py +24 -0
  304. mlflow/protos/unity_catalog_prompt_service_pb2.py +164 -0
  305. mlflow/protos/unity_catalog_prompt_service_pb2_grpc.py +785 -0
  306. mlflow/py.typed +0 -0
  307. mlflow/pydantic_ai/__init__.py +57 -0
  308. mlflow/pydantic_ai/autolog.py +173 -0
  309. mlflow/pyfunc/__init__.py +3844 -0
  310. mlflow/pyfunc/_mlflow_pyfunc_backend_predict.py +61 -0
  311. mlflow/pyfunc/backend.py +523 -0
  312. mlflow/pyfunc/context.py +78 -0
  313. mlflow/pyfunc/dbconnect_artifact_cache.py +144 -0
  314. mlflow/pyfunc/loaders/__init__.py +7 -0
  315. mlflow/pyfunc/loaders/chat_agent.py +117 -0
  316. mlflow/pyfunc/loaders/chat_model.py +125 -0
  317. mlflow/pyfunc/loaders/code_model.py +31 -0
  318. mlflow/pyfunc/loaders/responses_agent.py +112 -0
  319. mlflow/pyfunc/mlserver.py +46 -0
  320. mlflow/pyfunc/model.py +1473 -0
  321. mlflow/pyfunc/scoring_server/__init__.py +604 -0
  322. mlflow/pyfunc/scoring_server/app.py +7 -0
  323. mlflow/pyfunc/scoring_server/client.py +146 -0
  324. mlflow/pyfunc/spark_model_cache.py +48 -0
  325. mlflow/pyfunc/stdin_server.py +44 -0
  326. mlflow/pyfunc/utils/__init__.py +3 -0
  327. mlflow/pyfunc/utils/data_validation.py +224 -0
  328. mlflow/pyfunc/utils/environment.py +22 -0
  329. mlflow/pyfunc/utils/input_converter.py +47 -0
  330. mlflow/pyfunc/utils/serving_data_parser.py +11 -0
  331. mlflow/pytorch/__init__.py +1171 -0
  332. mlflow/pytorch/_lightning_autolog.py +580 -0
  333. mlflow/pytorch/_pytorch_autolog.py +50 -0
  334. mlflow/pytorch/pickle_module.py +35 -0
  335. mlflow/rfunc/__init__.py +42 -0
  336. mlflow/rfunc/backend.py +134 -0
  337. mlflow/runs.py +89 -0
  338. mlflow/server/__init__.py +302 -0
  339. mlflow/server/auth/__init__.py +1224 -0
  340. mlflow/server/auth/__main__.py +4 -0
  341. mlflow/server/auth/basic_auth.ini +6 -0
  342. mlflow/server/auth/cli.py +11 -0
  343. mlflow/server/auth/client.py +537 -0
  344. mlflow/server/auth/config.py +34 -0
  345. mlflow/server/auth/db/__init__.py +0 -0
  346. mlflow/server/auth/db/cli.py +18 -0
  347. mlflow/server/auth/db/migrations/__init__.py +0 -0
  348. mlflow/server/auth/db/migrations/alembic.ini +110 -0
  349. mlflow/server/auth/db/migrations/env.py +76 -0
  350. mlflow/server/auth/db/migrations/versions/8606fa83a998_initial_migration.py +51 -0
  351. mlflow/server/auth/db/migrations/versions/__init__.py +0 -0
  352. mlflow/server/auth/db/models.py +67 -0
  353. mlflow/server/auth/db/utils.py +37 -0
  354. mlflow/server/auth/entities.py +165 -0
  355. mlflow/server/auth/logo.py +14 -0
  356. mlflow/server/auth/permissions.py +65 -0
  357. mlflow/server/auth/routes.py +18 -0
  358. mlflow/server/auth/sqlalchemy_store.py +263 -0
  359. mlflow/server/graphql/__init__.py +0 -0
  360. mlflow/server/graphql/autogenerated_graphql_schema.py +353 -0
  361. mlflow/server/graphql/graphql_custom_scalars.py +24 -0
  362. mlflow/server/graphql/graphql_errors.py +15 -0
  363. mlflow/server/graphql/graphql_no_batching.py +89 -0
  364. mlflow/server/graphql/graphql_schema_extensions.py +74 -0
  365. mlflow/server/handlers.py +3217 -0
  366. mlflow/server/prometheus_exporter.py +17 -0
  367. mlflow/server/validation.py +30 -0
  368. mlflow/shap/__init__.py +691 -0
  369. mlflow/sklearn/__init__.py +1994 -0
  370. mlflow/sklearn/utils.py +1041 -0
  371. mlflow/smolagents/__init__.py +66 -0
  372. mlflow/smolagents/autolog.py +139 -0
  373. mlflow/smolagents/chat.py +29 -0
  374. mlflow/store/__init__.py +10 -0
  375. mlflow/store/_unity_catalog/__init__.py +1 -0
  376. mlflow/store/_unity_catalog/lineage/__init__.py +1 -0
  377. mlflow/store/_unity_catalog/lineage/constants.py +2 -0
  378. mlflow/store/_unity_catalog/registry/__init__.py +6 -0
  379. mlflow/store/_unity_catalog/registry/prompt_info.py +75 -0
  380. mlflow/store/_unity_catalog/registry/rest_store.py +1740 -0
  381. mlflow/store/_unity_catalog/registry/uc_oss_rest_store.py +507 -0
  382. mlflow/store/_unity_catalog/registry/utils.py +121 -0
  383. mlflow/store/artifact/__init__.py +0 -0
  384. mlflow/store/artifact/artifact_repo.py +472 -0
  385. mlflow/store/artifact/artifact_repository_registry.py +154 -0
  386. mlflow/store/artifact/azure_blob_artifact_repo.py +275 -0
  387. mlflow/store/artifact/azure_data_lake_artifact_repo.py +295 -0
  388. mlflow/store/artifact/cli.py +141 -0
  389. mlflow/store/artifact/cloud_artifact_repo.py +332 -0
  390. mlflow/store/artifact/databricks_artifact_repo.py +729 -0
  391. mlflow/store/artifact/databricks_artifact_repo_resources.py +301 -0
  392. mlflow/store/artifact/databricks_logged_model_artifact_repo.py +93 -0
  393. mlflow/store/artifact/databricks_models_artifact_repo.py +216 -0
  394. mlflow/store/artifact/databricks_sdk_artifact_repo.py +134 -0
  395. mlflow/store/artifact/databricks_sdk_models_artifact_repo.py +97 -0
  396. mlflow/store/artifact/dbfs_artifact_repo.py +240 -0
  397. mlflow/store/artifact/ftp_artifact_repo.py +132 -0
  398. mlflow/store/artifact/gcs_artifact_repo.py +296 -0
  399. mlflow/store/artifact/hdfs_artifact_repo.py +209 -0
  400. mlflow/store/artifact/http_artifact_repo.py +218 -0
  401. mlflow/store/artifact/local_artifact_repo.py +142 -0
  402. mlflow/store/artifact/mlflow_artifacts_repo.py +94 -0
  403. mlflow/store/artifact/models_artifact_repo.py +259 -0
  404. mlflow/store/artifact/optimized_s3_artifact_repo.py +356 -0
  405. mlflow/store/artifact/presigned_url_artifact_repo.py +173 -0
  406. mlflow/store/artifact/r2_artifact_repo.py +70 -0
  407. mlflow/store/artifact/runs_artifact_repo.py +265 -0
  408. mlflow/store/artifact/s3_artifact_repo.py +330 -0
  409. mlflow/store/artifact/sftp_artifact_repo.py +141 -0
  410. mlflow/store/artifact/uc_volume_artifact_repo.py +76 -0
  411. mlflow/store/artifact/unity_catalog_models_artifact_repo.py +168 -0
  412. mlflow/store/artifact/unity_catalog_oss_models_artifact_repo.py +168 -0
  413. mlflow/store/artifact/utils/__init__.py +0 -0
  414. mlflow/store/artifact/utils/models.py +148 -0
  415. mlflow/store/db/__init__.py +0 -0
  416. mlflow/store/db/base_sql_model.py +3 -0
  417. mlflow/store/db/db_types.py +10 -0
  418. mlflow/store/db/utils.py +314 -0
  419. mlflow/store/db_migrations/__init__.py +0 -0
  420. mlflow/store/db_migrations/alembic.ini +74 -0
  421. mlflow/store/db_migrations/env.py +84 -0
  422. mlflow/store/db_migrations/versions/0584bdc529eb_add_cascading_deletion_to_datasets_from_experiments.py +88 -0
  423. mlflow/store/db_migrations/versions/0a8213491aaa_drop_duplicate_killed_constraint.py +49 -0
  424. mlflow/store/db_migrations/versions/0c779009ac13_add_deleted_time_field_to_runs_table.py +24 -0
  425. mlflow/store/db_migrations/versions/181f10493468_allow_nulls_for_metric_values.py +35 -0
  426. mlflow/store/db_migrations/versions/27a6a02d2cf1_add_model_version_tags_table.py +38 -0
  427. mlflow/store/db_migrations/versions/2b4d017a5e9b_add_model_registry_tables_to_db.py +77 -0
  428. mlflow/store/db_migrations/versions/2d6e25af4d3e_increase_max_param_val_length.py +33 -0
  429. mlflow/store/db_migrations/versions/3500859a5d39_add_model_aliases_table.py +50 -0
  430. mlflow/store/db_migrations/versions/39d1c3be5f05_add_is_nan_constraint_for_metrics_tables_if_necessary.py +41 -0
  431. mlflow/store/db_migrations/versions/400f98739977_add_logged_model_tables.py +123 -0
  432. mlflow/store/db_migrations/versions/4465047574b1_increase_max_dataset_schema_size.py +38 -0
  433. mlflow/store/db_migrations/versions/451aebb31d03_add_metric_step.py +35 -0
  434. mlflow/store/db_migrations/versions/5b0e9adcef9c_add_cascade_deletion_to_trace_tables_fk.py +40 -0
  435. mlflow/store/db_migrations/versions/6953534de441_add_step_to_inputs_table.py +25 -0
  436. mlflow/store/db_migrations/versions/728d730b5ebd_add_registered_model_tags_table.py +38 -0
  437. mlflow/store/db_migrations/versions/7ac759974ad8_update_run_tags_with_larger_limit.py +36 -0
  438. mlflow/store/db_migrations/versions/7f2a7d5fae7d_add_datasets_inputs_input_tags_tables.py +82 -0
  439. mlflow/store/db_migrations/versions/84291f40a231_add_run_link_to_model_version.py +26 -0
  440. mlflow/store/db_migrations/versions/867495a8f9d4_add_trace_tables.py +90 -0
  441. mlflow/store/db_migrations/versions/89d4b8295536_create_latest_metrics_table.py +169 -0
  442. mlflow/store/db_migrations/versions/90e64c465722_migrate_user_column_to_tags.py +64 -0
  443. mlflow/store/db_migrations/versions/97727af70f4d_creation_time_last_update_time_experiments.py +25 -0
  444. mlflow/store/db_migrations/versions/__init__.py +0 -0
  445. mlflow/store/db_migrations/versions/a8c4a736bde6_allow_nulls_for_run_id.py +27 -0
  446. mlflow/store/db_migrations/versions/acf3f17fdcc7_add_storage_location_field_to_model_.py +29 -0
  447. mlflow/store/db_migrations/versions/bd07f7e963c5_create_index_on_run_uuid.py +26 -0
  448. mlflow/store/db_migrations/versions/bda7b8c39065_increase_model_version_tag_value_limit.py +38 -0
  449. mlflow/store/db_migrations/versions/c48cb773bb87_reset_default_value_for_is_nan_in_metrics_table_for_mysql.py +41 -0
  450. mlflow/store/db_migrations/versions/cbc13b556ace_add_v3_trace_schema_columns.py +31 -0
  451. mlflow/store/db_migrations/versions/cc1f77228345_change_param_value_length_to_500.py +34 -0
  452. mlflow/store/db_migrations/versions/cfd24bdc0731_update_run_status_constraint_with_killed.py +78 -0
  453. mlflow/store/db_migrations/versions/df50e92ffc5e_add_experiment_tags_table.py +38 -0
  454. mlflow/store/db_migrations/versions/f5a4f2784254_increase_run_tag_value_limit.py +36 -0
  455. mlflow/store/entities/__init__.py +3 -0
  456. mlflow/store/entities/paged_list.py +18 -0
  457. mlflow/store/model_registry/__init__.py +10 -0
  458. mlflow/store/model_registry/abstract_store.py +1081 -0
  459. mlflow/store/model_registry/base_rest_store.py +44 -0
  460. mlflow/store/model_registry/databricks_workspace_model_registry_rest_store.py +37 -0
  461. mlflow/store/model_registry/dbmodels/__init__.py +0 -0
  462. mlflow/store/model_registry/dbmodels/models.py +206 -0
  463. mlflow/store/model_registry/file_store.py +1091 -0
  464. mlflow/store/model_registry/rest_store.py +481 -0
  465. mlflow/store/model_registry/sqlalchemy_store.py +1286 -0
  466. mlflow/store/tracking/__init__.py +23 -0
  467. mlflow/store/tracking/abstract_store.py +816 -0
  468. mlflow/store/tracking/dbmodels/__init__.py +0 -0
  469. mlflow/store/tracking/dbmodels/initial_models.py +243 -0
  470. mlflow/store/tracking/dbmodels/models.py +1073 -0
  471. mlflow/store/tracking/file_store.py +2438 -0
  472. mlflow/store/tracking/postgres_managed_identity.py +146 -0
  473. mlflow/store/tracking/rest_store.py +1131 -0
  474. mlflow/store/tracking/sqlalchemy_store.py +2785 -0
  475. mlflow/system_metrics/__init__.py +61 -0
  476. mlflow/system_metrics/metrics/__init__.py +0 -0
  477. mlflow/system_metrics/metrics/base_metrics_monitor.py +32 -0
  478. mlflow/system_metrics/metrics/cpu_monitor.py +23 -0
  479. mlflow/system_metrics/metrics/disk_monitor.py +21 -0
  480. mlflow/system_metrics/metrics/gpu_monitor.py +71 -0
  481. mlflow/system_metrics/metrics/network_monitor.py +34 -0
  482. mlflow/system_metrics/metrics/rocm_monitor.py +123 -0
  483. mlflow/system_metrics/system_metrics_monitor.py +198 -0
  484. mlflow/tracing/__init__.py +16 -0
  485. mlflow/tracing/assessment.py +356 -0
  486. mlflow/tracing/client.py +531 -0
  487. mlflow/tracing/config.py +125 -0
  488. mlflow/tracing/constant.py +105 -0
  489. mlflow/tracing/destination.py +81 -0
  490. mlflow/tracing/display/__init__.py +40 -0
  491. mlflow/tracing/display/display_handler.py +196 -0
  492. mlflow/tracing/export/async_export_queue.py +186 -0
  493. mlflow/tracing/export/inference_table.py +138 -0
  494. mlflow/tracing/export/mlflow_v3.py +137 -0
  495. mlflow/tracing/export/utils.py +70 -0
  496. mlflow/tracing/fluent.py +1417 -0
  497. mlflow/tracing/processor/base_mlflow.py +199 -0
  498. mlflow/tracing/processor/inference_table.py +175 -0
  499. mlflow/tracing/processor/mlflow_v3.py +47 -0
  500. mlflow/tracing/processor/otel.py +73 -0
  501. mlflow/tracing/provider.py +487 -0
  502. mlflow/tracing/trace_manager.py +200 -0
  503. mlflow/tracing/utils/__init__.py +616 -0
  504. mlflow/tracing/utils/artifact_utils.py +28 -0
  505. mlflow/tracing/utils/copy.py +55 -0
  506. mlflow/tracing/utils/environment.py +55 -0
  507. mlflow/tracing/utils/exception.py +21 -0
  508. mlflow/tracing/utils/once.py +35 -0
  509. mlflow/tracing/utils/otlp.py +63 -0
  510. mlflow/tracing/utils/processor.py +54 -0
  511. mlflow/tracing/utils/search.py +292 -0
  512. mlflow/tracing/utils/timeout.py +250 -0
  513. mlflow/tracing/utils/token.py +19 -0
  514. mlflow/tracing/utils/truncation.py +124 -0
  515. mlflow/tracing/utils/warning.py +76 -0
  516. mlflow/tracking/__init__.py +39 -0
  517. mlflow/tracking/_model_registry/__init__.py +1 -0
  518. mlflow/tracking/_model_registry/client.py +764 -0
  519. mlflow/tracking/_model_registry/fluent.py +853 -0
  520. mlflow/tracking/_model_registry/registry.py +67 -0
  521. mlflow/tracking/_model_registry/utils.py +251 -0
  522. mlflow/tracking/_tracking_service/__init__.py +0 -0
  523. mlflow/tracking/_tracking_service/client.py +883 -0
  524. mlflow/tracking/_tracking_service/registry.py +56 -0
  525. mlflow/tracking/_tracking_service/utils.py +275 -0
  526. mlflow/tracking/artifact_utils.py +179 -0
  527. mlflow/tracking/client.py +5900 -0
  528. mlflow/tracking/context/__init__.py +0 -0
  529. mlflow/tracking/context/abstract_context.py +35 -0
  530. mlflow/tracking/context/databricks_cluster_context.py +15 -0
  531. mlflow/tracking/context/databricks_command_context.py +15 -0
  532. mlflow/tracking/context/databricks_job_context.py +49 -0
  533. mlflow/tracking/context/databricks_notebook_context.py +41 -0
  534. mlflow/tracking/context/databricks_repo_context.py +43 -0
  535. mlflow/tracking/context/default_context.py +51 -0
  536. mlflow/tracking/context/git_context.py +32 -0
  537. mlflow/tracking/context/registry.py +98 -0
  538. mlflow/tracking/context/system_environment_context.py +15 -0
  539. mlflow/tracking/default_experiment/__init__.py +1 -0
  540. mlflow/tracking/default_experiment/abstract_context.py +43 -0
  541. mlflow/tracking/default_experiment/databricks_notebook_experiment_provider.py +44 -0
  542. mlflow/tracking/default_experiment/registry.py +75 -0
  543. mlflow/tracking/fluent.py +3595 -0
  544. mlflow/tracking/metric_value_conversion_utils.py +93 -0
  545. mlflow/tracking/multimedia.py +206 -0
  546. mlflow/tracking/registry.py +86 -0
  547. mlflow/tracking/request_auth/__init__.py +0 -0
  548. mlflow/tracking/request_auth/abstract_request_auth_provider.py +34 -0
  549. mlflow/tracking/request_auth/registry.py +60 -0
  550. mlflow/tracking/request_header/__init__.py +0 -0
  551. mlflow/tracking/request_header/abstract_request_header_provider.py +36 -0
  552. mlflow/tracking/request_header/databricks_request_header_provider.py +38 -0
  553. mlflow/tracking/request_header/default_request_header_provider.py +17 -0
  554. mlflow/tracking/request_header/registry.py +79 -0
  555. mlflow/transformers/__init__.py +2982 -0
  556. mlflow/transformers/flavor_config.py +258 -0
  557. mlflow/transformers/hub_utils.py +83 -0
  558. mlflow/transformers/llm_inference_utils.py +468 -0
  559. mlflow/transformers/model_io.py +301 -0
  560. mlflow/transformers/peft.py +51 -0
  561. mlflow/transformers/signature.py +183 -0
  562. mlflow/transformers/torch_utils.py +55 -0
  563. mlflow/types/__init__.py +21 -0
  564. mlflow/types/agent.py +270 -0
  565. mlflow/types/chat.py +240 -0
  566. mlflow/types/llm.py +935 -0
  567. mlflow/types/responses.py +139 -0
  568. mlflow/types/responses_helpers.py +416 -0
  569. mlflow/types/schema.py +1505 -0
  570. mlflow/types/type_hints.py +647 -0
  571. mlflow/types/utils.py +753 -0
  572. mlflow/utils/__init__.py +283 -0
  573. mlflow/utils/_capture_modules.py +256 -0
  574. mlflow/utils/_capture_transformers_modules.py +75 -0
  575. mlflow/utils/_spark_utils.py +201 -0
  576. mlflow/utils/_unity_catalog_oss_utils.py +97 -0
  577. mlflow/utils/_unity_catalog_utils.py +479 -0
  578. mlflow/utils/annotations.py +218 -0
  579. mlflow/utils/arguments_utils.py +16 -0
  580. mlflow/utils/async_logging/__init__.py +1 -0
  581. mlflow/utils/async_logging/async_artifacts_logging_queue.py +258 -0
  582. mlflow/utils/async_logging/async_logging_queue.py +366 -0
  583. mlflow/utils/async_logging/run_artifact.py +38 -0
  584. mlflow/utils/async_logging/run_batch.py +58 -0
  585. mlflow/utils/async_logging/run_operations.py +49 -0
  586. mlflow/utils/autologging_utils/__init__.py +737 -0
  587. mlflow/utils/autologging_utils/client.py +432 -0
  588. mlflow/utils/autologging_utils/config.py +33 -0
  589. mlflow/utils/autologging_utils/events.py +294 -0
  590. mlflow/utils/autologging_utils/logging_and_warnings.py +328 -0
  591. mlflow/utils/autologging_utils/metrics_queue.py +71 -0
  592. mlflow/utils/autologging_utils/safety.py +1104 -0
  593. mlflow/utils/autologging_utils/versioning.py +95 -0
  594. mlflow/utils/checkpoint_utils.py +206 -0
  595. mlflow/utils/class_utils.py +6 -0
  596. mlflow/utils/cli_args.py +257 -0
  597. mlflow/utils/conda.py +354 -0
  598. mlflow/utils/credentials.py +231 -0
  599. mlflow/utils/data_utils.py +17 -0
  600. mlflow/utils/databricks_utils.py +1436 -0
  601. mlflow/utils/docstring_utils.py +477 -0
  602. mlflow/utils/doctor.py +133 -0
  603. mlflow/utils/download_cloud_file_chunk.py +43 -0
  604. mlflow/utils/env_manager.py +16 -0
  605. mlflow/utils/env_pack.py +131 -0
  606. mlflow/utils/environment.py +1009 -0
  607. mlflow/utils/exception_utils.py +14 -0
  608. mlflow/utils/file_utils.py +978 -0
  609. mlflow/utils/git_utils.py +77 -0
  610. mlflow/utils/gorilla.py +797 -0
  611. mlflow/utils/import_hooks/__init__.py +363 -0
  612. mlflow/utils/lazy_load.py +51 -0
  613. mlflow/utils/logging_utils.py +168 -0
  614. mlflow/utils/mime_type_utils.py +58 -0
  615. mlflow/utils/mlflow_tags.py +103 -0
  616. mlflow/utils/model_utils.py +486 -0
  617. mlflow/utils/name_utils.py +346 -0
  618. mlflow/utils/nfs_on_spark.py +62 -0
  619. mlflow/utils/openai_utils.py +164 -0
  620. mlflow/utils/os.py +12 -0
  621. mlflow/utils/oss_registry_utils.py +29 -0
  622. mlflow/utils/plugins.py +17 -0
  623. mlflow/utils/process.py +182 -0
  624. mlflow/utils/promptlab_utils.py +146 -0
  625. mlflow/utils/proto_json_utils.py +743 -0
  626. mlflow/utils/pydantic_utils.py +54 -0
  627. mlflow/utils/request_utils.py +279 -0
  628. mlflow/utils/requirements_utils.py +704 -0
  629. mlflow/utils/rest_utils.py +673 -0
  630. mlflow/utils/search_logged_model_utils.py +127 -0
  631. mlflow/utils/search_utils.py +2111 -0
  632. mlflow/utils/secure_loading.py +221 -0
  633. mlflow/utils/security_validation.py +384 -0
  634. mlflow/utils/server_cli_utils.py +61 -0
  635. mlflow/utils/spark_utils.py +15 -0
  636. mlflow/utils/string_utils.py +138 -0
  637. mlflow/utils/thread_utils.py +63 -0
  638. mlflow/utils/time.py +54 -0
  639. mlflow/utils/timeout.py +42 -0
  640. mlflow/utils/uri.py +572 -0
  641. mlflow/utils/validation.py +662 -0
  642. mlflow/utils/virtualenv.py +458 -0
  643. mlflow/utils/warnings_utils.py +25 -0
  644. mlflow/utils/yaml_utils.py +179 -0
  645. mlflow/version.py +24 -0
mlflow/protos/databricks_uc_registry_service_pb2.py ADDED
@@ -0,0 +1,170 @@
1
+
2
+ import google.protobuf
3
+ from packaging.version import Version
4
+ if Version(google.protobuf.__version__).major >= 5:
5
+ # -*- coding: utf-8 -*-
6
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
7
+ # source: databricks_uc_registry_service.proto
8
+ # Protobuf Python Version: 5.26.0
9
+ """Generated protocol buffer code."""
10
+ from google.protobuf import descriptor as _descriptor
11
+ from google.protobuf import descriptor_pool as _descriptor_pool
12
+ from google.protobuf import symbol_database as _symbol_database
13
+ from google.protobuf.internal import builder as _builder
14
+ # @@protoc_insertion_point(imports)
15
+
16
+ _sym_db = _symbol_database.Default()
17
+
18
+
19
+ from . import databricks_pb2 as databricks__pb2
20
+ from . import databricks_uc_registry_messages_pb2 as databricks_uc_registry_messages_pb2
21
+ from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
22
+
23
+
24
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$databricks_uc_registry_service.proto\x12\x16mlflow.ucmodelregistry\x1a\x10\x64\x61tabricks.proto\x1a%databricks_uc_registry_messages.proto\x1a\x15scalapb/scalapb.proto2\xc0,\n\x16UcModelRegistryService\x12\xf2\x01\n\x15\x43reateRegisteredModel\x12\x34.mlflow.ucmodelregistry.CreateRegisteredModelRequest\x1a\x35.mlflow.ucmodelregistry.CreateRegisteredModelResponse\"l\xf2\x86\x19h\n<\n\x04POST\x12./mlflow/unity-catalog/registered-models/create\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Create RegisteredModel\x12\xf3\x01\n\x15UpdateRegisteredModel\x12\x34.mlflow.ucmodelregistry.UpdateRegisteredModelRequest\x1a\x35.mlflow.ucmodelregistry.UpdateRegisteredModelResponse\"m\xf2\x86\x19i\n=\n\x05PATCH\x12./mlflow/unity-catalog/registered-models/update\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Update RegisteredModel\x12\xf4\x01\n\x15\x44\x65leteRegisteredModel\x12\x34.mlflow.ucmodelregistry.DeleteRegisteredModelRequest\x1a\x35.mlflow.ucmodelregistry.DeleteRegisteredModelResponse\"n\xf2\x86\x19j\n>\n\x06\x44\x45LETE\x12./mlflow/unity-catalog/registered-models/delete\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Delete RegisteredModel\x12\xe2\x01\n\x12GetRegisteredModel\x12\x31.mlflow.ucmodelregistry.GetRegisteredModelRequest\x1a\x32.mlflow.ucmodelregistry.GetRegisteredModelResponse\"e\xf2\x86\x19\x61\n8\n\x03GET\x12+/mlflow/unity-catalog/registered-models/get\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Get RegisteredModel\x12\xf5\x01\n\x16SearchRegisteredModels\x12\x35.mlflow.ucmodelregistry.SearchRegisteredModelsRequest\x1a\x36.mlflow.ucmodelregistry.SearchRegisteredModelsResponse\"l\xf2\x86\x19h\n;\n\x03GET\x12./mlflow/unity-catalog/registered-models/search\x1a\x04\x08\x02\x10\x00\x10\x03*\'(Unity Catalog) Search RegisteredModels\x12\xe3\x01\n\x12\x43reateModelVersion\x12\x31.mlflow.ucmodelregistry.CreateModelVersionRequest\x1a\x32.mlflow.ucmodelregistry.CreateModelVersionResponse\"f\xf2\x86\x19\x62\n9\n\x04POST\x12+/mlflow/unity-catalog/model-versions/create\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Create ModelVersion\x12\xe4\x01\n\x12UpdateModelVersion\x12\x31.mlflow.ucmodelregistry.UpdateModelVersionRequest\x1a\x32.mlflow.ucmodelregistry.UpdateModelVersionResponse\"g\xf2\x86\x19\x63\n:\n\x05PATCH\x12+/mlflow/unity-catalog/model-versions/update\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Update ModelVersion\x12\xe5\x01\n\x12\x44\x65leteModelVersion\x12\x31.mlflow.ucmodelregistry.DeleteModelVersionRequest\x1a\x32.mlflow.ucmodelregistry.DeleteModelVersionResponse\"h\xf2\x86\x19\x64\n;\n\x06\x44\x45LETE\x12+/mlflow/unity-catalog/model-versions/delete\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Delete ModelVersion\x12\xd3\x01\n\x0fGetModelVersion\x12..mlflow.ucmodelregistry.GetModelVersionRequest\x1a/.mlflow.ucmodelregistry.GetModelVersionResponse\"_\xf2\x86\x19[\n5\n\x03GET\x12(/mlflow/unity-catalog/model-versions/get\x1a\x04\x08\x02\x10\x00\x10\x03* (Unity Catalog) Get ModelVersion\x12\xe6\x01\n\x13SearchModelVersions\x12\x32.mlflow.ucmodelregistry.SearchModelVersionsRequest\x1a\x33.mlflow.ucmodelregistry.SearchModelVersionsResponse\"f\xf2\x86\x19\x62\n8\n\x03GET\x12+/mlflow/unity-catalog/model-versions/search\x1a\x04\x08\x02\x10\x00\x10\x03*$(Unity Catalog) Search ModelVersions\x12\xd8\x02\n(GenerateTemporaryModelVersionCredentials\x12G.mlflow.ucmodelregistry.GenerateTemporaryModelVersionCredentialsRequest\x1aH.mlflow.ucmodelregistry.GenerateTemporaryModelVersionCredentialsResponse\"\x98\x01\xf2\x86\x19\x93\x01\nQ\n\x04POST\x12\x43/mlflow/unity-catalog/model-versions/generate-temporary-credentials\x1a\x04\x08\x02\x10\x00\x10\x03*<(Unity Catalog) Generate Temporary Model Version Credentials\x12\x9e\x02\n\x1aGetModelVersionDownloadUri\x12\x39.mlflow.ucmodelregistry.GetModelVersionDownloadUriRequest\x1a:.mlflow.ucmodelregistry.GetModelVersionDownloadUriResponse\"\x88\x01\xf2\x86\x19\x83\x01\nB\n\x03GET\x12\x35/mlflow/unity-catalog/model-versions/get-download-uri\x1a\x04\x08\x02\x10\x00\x10\x03*;(Unity Catalog) Get Download URI For ModelVersion Artifacts\x12\xee\x01\n\x14\x46inalizeModelVersion\x12\x33.mlflow.ucmodelregistry.FinalizeModelVersionRequest\x1a\x34.mlflow.ucmodelregistry.FinalizeModelVersionResponse\"k\xf2\x86\x19g\n;\n\x04POST\x12-/mlflow/unity-catalog/model-versions/finalize\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Finalize Model Version\x12\xfa\x01\n\x17SetRegisteredModelAlias\x12\x36.mlflow.ucmodelregistry.SetRegisteredModelAliasRequest\x1a\x37.mlflow.ucmodelregistry.SetRegisteredModelAliasResponse\"n\xf2\x86\x19j\n;\n\x04POST\x12-/mlflow/unity-catalog/registered-models/alias\x1a\x04\x08\x02\x10\x00\x10\x03*)(Unity Catalog) Set RegisteredModel Alias\x12\x88\x02\n\x1a\x44\x65leteRegisteredModelAlias\x12\x39.mlflow.ucmodelregistry.DeleteRegisteredModelAliasRequest\x1a:.mlflow.ucmodelregistry.DeleteRegisteredModelAliasResponse\"s\xf2\x86\x19o\n=\n\x06\x44\x45LETE\x12-/mlflow/unity-catalog/registered-models/alias\x1a\x04\x08\x02\x10\x00\x10\x03*,(Unity Catalog) Delete RegisteredModel Alias\x12\xf6\x01\n\x16GetModelVersionByAlias\x12\x35.mlflow.ucmodelregistry.GetModelVersionByAliasRequest\x1a\x36.mlflow.ucmodelregistry.GetModelVersionByAliasResponse\"m\xf2\x86\x19i\n:\n\x03GET\x12-/mlflow/unity-catalog/registered-models/alias\x1a\x04\x08\x02\x10\x00\x10\x03*)(Unity Catalog) Get ModelVersion By Alias\x12\xf4\x01\n\x15SetRegisteredModelTag\x12\x34.mlflow.ucmodelregistry.SetRegisteredModelTagRequest\x1a\x35.mlflow.ucmodelregistry.SetRegisteredModelTagResponse\"n\xf2\x86\x19j\n=\n\x04POST\x12//mlflow/unity-catalog/registered-models/set-tag\x1a\x04\x08\x02\x10\x00\x10\x03*\'(Unity Catalog) Set RegisteredModel Tag\x12\x85\x02\n\x18\x44\x65leteRegisteredModelTag\x12\x37.mlflow.ucmodelregistry.DeleteRegisteredModelTagRequest\x1a\x38.mlflow.ucmodelregistry.DeleteRegisteredModelTagResponse\"v\xf2\x86\x19r\nB\n\x06\x44\x45LETE\x12\x32/mlflow/unity-catalog/registered-models/delete-tag\x1a\x04\x08\x02\x10\x00\x10\x03**(Unity Catalog) Delete RegisteredModel Tag\x12\xe5\x01\n\x12SetModelVersionTag\x12\x31.mlflow.ucmodelregistry.SetModelVersionTagRequest\x1a\x32.mlflow.ucmodelregistry.SetModelVersionTagResponse\"h\xf2\x86\x19\x64\n:\n\x04POST\x12,/mlflow/unity-catalog/model-versions/set-tag\x1a\x04\x08\x02\x10\x00\x10\x03*$(Unity Catalog) Set ModelVersion Tag\x12\xf6\x01\n\x15\x44\x65leteModelVersionTag\x12\x34.mlflow.ucmodelregistry.DeleteModelVersionTagRequest\x1a\x35.mlflow.ucmodelregistry.DeleteModelVersionTagResponse\"p\xf2\x86\x19l\n?\n\x06\x44\x45LETE\x12//mlflow/unity-catalog/model-versions/delete-tag\x1a\x04\x08\x02\x10\x00\x10\x03*\'(Unity Catalog) Delete ModelVersion Tag\x12\xfe\x01\n\x17\x45mitModelVersionLineage\x12\x36.mlflow.ucmodelregistry.EmitModelVersionLineageRequest\x1a\x37.mlflow.ucmodelregistry.EmitModelVersionLineageResponse\"r\xf2\x86\x19n\n?\n\x04POST\x12\x31/mlflow/unity-catalog/model-versions/emit-lineage\x1a\x04\x08\x02\x10\x00\x10\x03*)(Unity Catalog) Emit ModelVersion Lineage\x12\xa3\x03\n.IsDatabricksSdkModelsArtifactRepositoryEnabled\x12M.mlflow.ucmodelregistry.IsDatabricksSdkModelsArtifactRepositoryEnabledRequest\x1aN.mlflow.ucmodelregistry.IsDatabricksSdkModelsArtifactRepositoryEnabledResponse\"\xd1\x01\xf2\x86\x19\xcc\x01\ni\n\x03GET\x12\\/mlflow/unity-catalog/registered-models:is-databricks-sdk-models-artifact-repository-enabled\x1a\x04\x08\x02\x10\x00\x10\x03*](Unity Catalog) Use DatabricksSdkModelsArtifactRepository in mlflow client for model registryB5\n(com.databricks.api.proto.ucmodelregistry\x90\x01\x01\xa0\x01\x01\xe2?\x02\x10\x01')
25
+
26
+ _globals = globals()
27
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
28
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'databricks_uc_registry_service_pb2', _globals)
29
+ if not _descriptor._USE_C_DESCRIPTORS:
30
+ _globals['DESCRIPTOR']._loaded_options = None
31
+ _globals['DESCRIPTOR']._serialized_options = b'\n(com.databricks.api.proto.ucmodelregistry\220\001\001\240\001\001\342?\002\020\001'
32
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['CreateRegisteredModel']._loaded_options = None
33
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['CreateRegisteredModel']._serialized_options = b'\362\206\031h\n<\n\004POST\022./mlflow/unity-catalog/registered-models/create\032\004\010\002\020\000\020\003*&(Unity Catalog) Create RegisteredModel'
34
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['UpdateRegisteredModel']._loaded_options = None
35
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['UpdateRegisteredModel']._serialized_options = b'\362\206\031i\n=\n\005PATCH\022./mlflow/unity-catalog/registered-models/update\032\004\010\002\020\000\020\003*&(Unity Catalog) Update RegisteredModel'
36
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteRegisteredModel']._loaded_options = None
37
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteRegisteredModel']._serialized_options = b'\362\206\031j\n>\n\006DELETE\022./mlflow/unity-catalog/registered-models/delete\032\004\010\002\020\000\020\003*&(Unity Catalog) Delete RegisteredModel'
38
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetRegisteredModel']._loaded_options = None
39
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetRegisteredModel']._serialized_options = b'\362\206\031a\n8\n\003GET\022+/mlflow/unity-catalog/registered-models/get\032\004\010\002\020\000\020\003*#(Unity Catalog) Get RegisteredModel'
40
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SearchRegisteredModels']._loaded_options = None
41
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SearchRegisteredModels']._serialized_options = b'\362\206\031h\n;\n\003GET\022./mlflow/unity-catalog/registered-models/search\032\004\010\002\020\000\020\003*\'(Unity Catalog) Search RegisteredModels'
42
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['CreateModelVersion']._loaded_options = None
43
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['CreateModelVersion']._serialized_options = b'\362\206\031b\n9\n\004POST\022+/mlflow/unity-catalog/model-versions/create\032\004\010\002\020\000\020\003*#(Unity Catalog) Create ModelVersion'
44
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['UpdateModelVersion']._loaded_options = None
45
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['UpdateModelVersion']._serialized_options = b'\362\206\031c\n:\n\005PATCH\022+/mlflow/unity-catalog/model-versions/update\032\004\010\002\020\000\020\003*#(Unity Catalog) Update ModelVersion'
46
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteModelVersion']._loaded_options = None
47
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteModelVersion']._serialized_options = b'\362\206\031d\n;\n\006DELETE\022+/mlflow/unity-catalog/model-versions/delete\032\004\010\002\020\000\020\003*#(Unity Catalog) Delete ModelVersion'
48
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetModelVersion']._loaded_options = None
49
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetModelVersion']._serialized_options = b'\362\206\031[\n5\n\003GET\022(/mlflow/unity-catalog/model-versions/get\032\004\010\002\020\000\020\003* (Unity Catalog) Get ModelVersion'
50
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SearchModelVersions']._loaded_options = None
51
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SearchModelVersions']._serialized_options = b'\362\206\031b\n8\n\003GET\022+/mlflow/unity-catalog/model-versions/search\032\004\010\002\020\000\020\003*$(Unity Catalog) Search ModelVersions'
52
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GenerateTemporaryModelVersionCredentials']._loaded_options = None
53
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GenerateTemporaryModelVersionCredentials']._serialized_options = b'\362\206\031\223\001\nQ\n\004POST\022C/mlflow/unity-catalog/model-versions/generate-temporary-credentials\032\004\010\002\020\000\020\003*<(Unity Catalog) Generate Temporary Model Version Credentials'
54
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetModelVersionDownloadUri']._loaded_options = None
55
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetModelVersionDownloadUri']._serialized_options = b'\362\206\031\203\001\nB\n\003GET\0225/mlflow/unity-catalog/model-versions/get-download-uri\032\004\010\002\020\000\020\003*;(Unity Catalog) Get Download URI For ModelVersion Artifacts'
56
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['FinalizeModelVersion']._loaded_options = None
57
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['FinalizeModelVersion']._serialized_options = b'\362\206\031g\n;\n\004POST\022-/mlflow/unity-catalog/model-versions/finalize\032\004\010\002\020\000\020\003*&(Unity Catalog) Finalize Model Version'
58
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SetRegisteredModelAlias']._loaded_options = None
59
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SetRegisteredModelAlias']._serialized_options = b'\362\206\031j\n;\n\004POST\022-/mlflow/unity-catalog/registered-models/alias\032\004\010\002\020\000\020\003*)(Unity Catalog) Set RegisteredModel Alias'
60
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteRegisteredModelAlias']._loaded_options = None
61
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteRegisteredModelAlias']._serialized_options = b'\362\206\031o\n=\n\006DELETE\022-/mlflow/unity-catalog/registered-models/alias\032\004\010\002\020\000\020\003*,(Unity Catalog) Delete RegisteredModel Alias'
62
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetModelVersionByAlias']._loaded_options = None
63
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['GetModelVersionByAlias']._serialized_options = b'\362\206\031i\n:\n\003GET\022-/mlflow/unity-catalog/registered-models/alias\032\004\010\002\020\000\020\003*)(Unity Catalog) Get ModelVersion By Alias'
64
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SetRegisteredModelTag']._loaded_options = None
65
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SetRegisteredModelTag']._serialized_options = b'\362\206\031j\n=\n\004POST\022//mlflow/unity-catalog/registered-models/set-tag\032\004\010\002\020\000\020\003*\'(Unity Catalog) Set RegisteredModel Tag'
66
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteRegisteredModelTag']._loaded_options = None
67
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteRegisteredModelTag']._serialized_options = b'\362\206\031r\nB\n\006DELETE\0222/mlflow/unity-catalog/registered-models/delete-tag\032\004\010\002\020\000\020\003**(Unity Catalog) Delete RegisteredModel Tag'
68
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SetModelVersionTag']._loaded_options = None
69
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['SetModelVersionTag']._serialized_options = b'\362\206\031d\n:\n\004POST\022,/mlflow/unity-catalog/model-versions/set-tag\032\004\010\002\020\000\020\003*$(Unity Catalog) Set ModelVersion Tag'
70
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteModelVersionTag']._loaded_options = None
71
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['DeleteModelVersionTag']._serialized_options = b'\362\206\031l\n?\n\006DELETE\022//mlflow/unity-catalog/model-versions/delete-tag\032\004\010\002\020\000\020\003*\'(Unity Catalog) Delete ModelVersion Tag'
72
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['EmitModelVersionLineage']._loaded_options = None
73
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['EmitModelVersionLineage']._serialized_options = b'\362\206\031n\n?\n\004POST\0221/mlflow/unity-catalog/model-versions/emit-lineage\032\004\010\002\020\000\020\003*)(Unity Catalog) Emit ModelVersion Lineage'
74
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['IsDatabricksSdkModelsArtifactRepositoryEnabled']._loaded_options = None
75
+ _globals['_UCMODELREGISTRYSERVICE'].methods_by_name['IsDatabricksSdkModelsArtifactRepositoryEnabled']._serialized_options = b'\362\206\031\314\001\ni\n\003GET\022\\/mlflow/unity-catalog/registered-models:is-databricks-sdk-models-artifact-repository-enabled\032\004\010\002\020\000\020\003*](Unity Catalog) Use DatabricksSdkModelsArtifactRepository in mlflow client for model registry'
76
+ _globals['_UCMODELREGISTRYSERVICE']._serialized_start=145
77
+ _globals['_UCMODELREGISTRYSERVICE']._serialized_end=5841
78
+ _builder.BuildServices(DESCRIPTOR, 'databricks_uc_registry_service_pb2', _globals)
79
+ # @@protoc_insertion_point(module_scope)
80
+
81
+ else:
82
+ # -*- coding: utf-8 -*-
83
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
84
+ # source: databricks_uc_registry_service.proto
85
+ """Generated protocol buffer code."""
86
+ from google.protobuf import descriptor as _descriptor
87
+ from google.protobuf import descriptor_pool as _descriptor_pool
88
+ from google.protobuf import message as _message
89
+ from google.protobuf import reflection as _reflection
90
+ from google.protobuf import symbol_database as _symbol_database
91
+ from google.protobuf import service as _service
92
+ from google.protobuf import service_reflection
93
+ # @@protoc_insertion_point(imports)
94
+
95
+ _sym_db = _symbol_database.Default()
96
+
97
+
98
+ from . import databricks_pb2 as databricks__pb2
99
+ from . import databricks_uc_registry_messages_pb2 as databricks_uc_registry_messages_pb2
100
+ from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
101
+
102
+
103
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$databricks_uc_registry_service.proto\x12\x16mlflow.ucmodelregistry\x1a\x10\x64\x61tabricks.proto\x1a%databricks_uc_registry_messages.proto\x1a\x15scalapb/scalapb.proto2\xc0,\n\x16UcModelRegistryService\x12\xf2\x01\n\x15\x43reateRegisteredModel\x12\x34.mlflow.ucmodelregistry.CreateRegisteredModelRequest\x1a\x35.mlflow.ucmodelregistry.CreateRegisteredModelResponse\"l\xf2\x86\x19h\n<\n\x04POST\x12./mlflow/unity-catalog/registered-models/create\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Create RegisteredModel\x12\xf3\x01\n\x15UpdateRegisteredModel\x12\x34.mlflow.ucmodelregistry.UpdateRegisteredModelRequest\x1a\x35.mlflow.ucmodelregistry.UpdateRegisteredModelResponse\"m\xf2\x86\x19i\n=\n\x05PATCH\x12./mlflow/unity-catalog/registered-models/update\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Update RegisteredModel\x12\xf4\x01\n\x15\x44\x65leteRegisteredModel\x12\x34.mlflow.ucmodelregistry.DeleteRegisteredModelRequest\x1a\x35.mlflow.ucmodelregistry.DeleteRegisteredModelResponse\"n\xf2\x86\x19j\n>\n\x06\x44\x45LETE\x12./mlflow/unity-catalog/registered-models/delete\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Delete RegisteredModel\x12\xe2\x01\n\x12GetRegisteredModel\x12\x31.mlflow.ucmodelregistry.GetRegisteredModelRequest\x1a\x32.mlflow.ucmodelregistry.GetRegisteredModelResponse\"e\xf2\x86\x19\x61\n8\n\x03GET\x12+/mlflow/unity-catalog/registered-models/get\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Get RegisteredModel\x12\xf5\x01\n\x16SearchRegisteredModels\x12\x35.mlflow.ucmodelregistry.SearchRegisteredModelsRequest\x1a\x36.mlflow.ucmodelregistry.SearchRegisteredModelsResponse\"l\xf2\x86\x19h\n;\n\x03GET\x12./mlflow/unity-catalog/registered-models/search\x1a\x04\x08\x02\x10\x00\x10\x03*\'(Unity Catalog) Search RegisteredModels\x12\xe3\x01\n\x12\x43reateModelVersion\x12\x31.mlflow.ucmodelregistry.CreateModelVersionRequest\x1a\x32.mlflow.ucmodelregistry.CreateModelVersionResponse\"f\xf2\x86\x19\x62\n9\n\x04POST\x12+/mlflow/unity-catalog/model-versions/create\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Create ModelVersion\x12\xe4\x01\n\x12UpdateModelVersion\x12\x31.mlflow.ucmodelregistry.UpdateModelVersionRequest\x1a\x32.mlflow.ucmodelregistry.UpdateModelVersionResponse\"g\xf2\x86\x19\x63\n:\n\x05PATCH\x12+/mlflow/unity-catalog/model-versions/update\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Update ModelVersion\x12\xe5\x01\n\x12\x44\x65leteModelVersion\x12\x31.mlflow.ucmodelregistry.DeleteModelVersionRequest\x1a\x32.mlflow.ucmodelregistry.DeleteModelVersionResponse\"h\xf2\x86\x19\x64\n;\n\x06\x44\x45LETE\x12+/mlflow/unity-catalog/model-versions/delete\x1a\x04\x08\x02\x10\x00\x10\x03*#(Unity Catalog) Delete ModelVersion\x12\xd3\x01\n\x0fGetModelVersion\x12..mlflow.ucmodelregistry.GetModelVersionRequest\x1a/.mlflow.ucmodelregistry.GetModelVersionResponse\"_\xf2\x86\x19[\n5\n\x03GET\x12(/mlflow/unity-catalog/model-versions/get\x1a\x04\x08\x02\x10\x00\x10\x03* (Unity Catalog) Get ModelVersion\x12\xe6\x01\n\x13SearchModelVersions\x12\x32.mlflow.ucmodelregistry.SearchModelVersionsRequest\x1a\x33.mlflow.ucmodelregistry.SearchModelVersionsResponse\"f\xf2\x86\x19\x62\n8\n\x03GET\x12+/mlflow/unity-catalog/model-versions/search\x1a\x04\x08\x02\x10\x00\x10\x03*$(Unity Catalog) Search ModelVersions\x12\xd8\x02\n(GenerateTemporaryModelVersionCredentials\x12G.mlflow.ucmodelregistry.GenerateTemporaryModelVersionCredentialsRequest\x1aH.mlflow.ucmodelregistry.GenerateTemporaryModelVersionCredentialsResponse\"\x98\x01\xf2\x86\x19\x93\x01\nQ\n\x04POST\x12\x43/mlflow/unity-catalog/model-versions/generate-temporary-credentials\x1a\x04\x08\x02\x10\x00\x10\x03*<(Unity Catalog) Generate Temporary Model Version Credentials\x12\x9e\x02\n\x1aGetModelVersionDownloadUri\x12\x39.mlflow.ucmodelregistry.GetModelVersionDownloadUriRequest\x1a:.mlflow.ucmodelregistry.GetModelVersionDownloadUriResponse\"\x88\x01\xf2\x86\x19\x83\x01\nB\n\x03GET\x12\x35/mlflow/unity-catalog/model-versions/get-download-uri\x1a\x04\x08\x02\x10\x00\x10\x03*;(Unity Catalog) Get Download URI For ModelVersion Artifacts\x12\xee\x01\n\x14\x46inalizeModelVersion\x12\x33.mlflow.ucmodelregistry.FinalizeModelVersionRequest\x1a\x34.mlflow.ucmodelregistry.FinalizeModelVersionResponse\"k\xf2\x86\x19g\n;\n\x04POST\x12-/mlflow/unity-catalog/model-versions/finalize\x1a\x04\x08\x02\x10\x00\x10\x03*&(Unity Catalog) Finalize Model Version\x12\xfa\x01\n\x17SetRegisteredModelAlias\x12\x36.mlflow.ucmodelregistry.SetRegisteredModelAliasRequest\x1a\x37.mlflow.ucmodelregistry.SetRegisteredModelAliasResponse\"n\xf2\x86\x19j\n;\n\x04POST\x12-/mlflow/unity-catalog/registered-models/alias\x1a\x04\x08\x02\x10\x00\x10\x03*)(Unity Catalog) Set RegisteredModel Alias\x12\x88\x02\n\x1a\x44\x65leteRegisteredModelAlias\x12\x39.mlflow.ucmodelregistry.DeleteRegisteredModelAliasRequest\x1a:.mlflow.ucmodelregistry.DeleteRegisteredModelAliasResponse\"s\xf2\x86\x19o\n=\n\x06\x44\x45LETE\x12-/mlflow/unity-catalog/registered-models/alias\x1a\x04\x08\x02\x10\x00\x10\x03*,(Unity Catalog) Delete RegisteredModel Alias\x12\xf6\x01\n\x16GetModelVersionByAlias\x12\x35.mlflow.ucmodelregistry.GetModelVersionByAliasRequest\x1a\x36.mlflow.ucmodelregistry.GetModelVersionByAliasResponse\"m\xf2\x86\x19i\n:\n\x03GET\x12-/mlflow/unity-catalog/registered-models/alias\x1a\x04\x08\x02\x10\x00\x10\x03*)(Unity Catalog) Get ModelVersion By Alias\x12\xf4\x01\n\x15SetRegisteredModelTag\x12\x34.mlflow.ucmodelregistry.SetRegisteredModelTagRequest\x1a\x35.mlflow.ucmodelregistry.SetRegisteredModelTagResponse\"n\xf2\x86\x19j\n=\n\x04POST\x12//mlflow/unity-catalog/registered-models/set-tag\x1a\x04\x08\x02\x10\x00\x10\x03*\'(Unity Catalog) Set RegisteredModel Tag\x12\x85\x02\n\x18\x44\x65leteRegisteredModelTag\x12\x37.mlflow.ucmodelregistry.DeleteRegisteredModelTagRequest\x1a\x38.mlflow.ucmodelregistry.DeleteRegisteredModelTagResponse\"v\xf2\x86\x19r\nB\n\x06\x44\x45LETE\x12\x32/mlflow/unity-catalog/registered-models/delete-tag\x1a\x04\x08\x02\x10\x00\x10\x03**(Unity Catalog) Delete RegisteredModel Tag\x12\xe5\x01\n\x12SetModelVersionTag\x12\x31.mlflow.ucmodelregistry.SetModelVersionTagRequest\x1a\x32.mlflow.ucmodelregistry.SetModelVersionTagResponse\"h\xf2\x86\x19\x64\n:\n\x04POST\x12,/mlflow/unity-catalog/model-versions/set-tag\x1a\x04\x08\x02\x10\x00\x10\x03*$(Unity Catalog) Set ModelVersion Tag\x12\xf6\x01\n\x15\x44\x65leteModelVersionTag\x12\x34.mlflow.ucmodelregistry.DeleteModelVersionTagRequest\x1a\x35.mlflow.ucmodelregistry.DeleteModelVersionTagResponse\"p\xf2\x86\x19l\n?\n\x06\x44\x45LETE\x12//mlflow/unity-catalog/model-versions/delete-tag\x1a\x04\x08\x02\x10\x00\x10\x03*\'(Unity Catalog) Delete ModelVersion Tag\x12\xfe\x01\n\x17\x45mitModelVersionLineage\x12\x36.mlflow.ucmodelregistry.EmitModelVersionLineageRequest\x1a\x37.mlflow.ucmodelregistry.EmitModelVersionLineageResponse\"r\xf2\x86\x19n\n?\n\x04POST\x12\x31/mlflow/unity-catalog/model-versions/emit-lineage\x1a\x04\x08\x02\x10\x00\x10\x03*)(Unity Catalog) Emit ModelVersion Lineage\x12\xa3\x03\n.IsDatabricksSdkModelsArtifactRepositoryEnabled\x12M.mlflow.ucmodelregistry.IsDatabricksSdkModelsArtifactRepositoryEnabledRequest\x1aN.mlflow.ucmodelregistry.IsDatabricksSdkModelsArtifactRepositoryEnabledResponse\"\xd1\x01\xf2\x86\x19\xcc\x01\ni\n\x03GET\x12\\/mlflow/unity-catalog/registered-models:is-databricks-sdk-models-artifact-repository-enabled\x1a\x04\x08\x02\x10\x00\x10\x03*](Unity Catalog) Use DatabricksSdkModelsArtifactRepository in mlflow client for model registryB5\n(com.databricks.api.proto.ucmodelregistry\x90\x01\x01\xa0\x01\x01\xe2?\x02\x10\x01')
104
+
105
+
106
+
107
+ _UCMODELREGISTRYSERVICE = DESCRIPTOR.services_by_name['UcModelRegistryService']
108
+ if _descriptor._USE_C_DESCRIPTORS == False:
109
+
110
+ DESCRIPTOR._options = None
111
+ DESCRIPTOR._serialized_options = b'\n(com.databricks.api.proto.ucmodelregistry\220\001\001\240\001\001\342?\002\020\001'
112
+ _UCMODELREGISTRYSERVICE.methods_by_name['CreateRegisteredModel']._options = None
113
+ _UCMODELREGISTRYSERVICE.methods_by_name['CreateRegisteredModel']._serialized_options = b'\362\206\031h\n<\n\004POST\022./mlflow/unity-catalog/registered-models/create\032\004\010\002\020\000\020\003*&(Unity Catalog) Create RegisteredModel'
114
+ _UCMODELREGISTRYSERVICE.methods_by_name['UpdateRegisteredModel']._options = None
115
+ _UCMODELREGISTRYSERVICE.methods_by_name['UpdateRegisteredModel']._serialized_options = b'\362\206\031i\n=\n\005PATCH\022./mlflow/unity-catalog/registered-models/update\032\004\010\002\020\000\020\003*&(Unity Catalog) Update RegisteredModel'
116
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteRegisteredModel']._options = None
117
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteRegisteredModel']._serialized_options = b'\362\206\031j\n>\n\006DELETE\022./mlflow/unity-catalog/registered-models/delete\032\004\010\002\020\000\020\003*&(Unity Catalog) Delete RegisteredModel'
118
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetRegisteredModel']._options = None
119
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetRegisteredModel']._serialized_options = b'\362\206\031a\n8\n\003GET\022+/mlflow/unity-catalog/registered-models/get\032\004\010\002\020\000\020\003*#(Unity Catalog) Get RegisteredModel'
120
+ _UCMODELREGISTRYSERVICE.methods_by_name['SearchRegisteredModels']._options = None
121
+ _UCMODELREGISTRYSERVICE.methods_by_name['SearchRegisteredModels']._serialized_options = b'\362\206\031h\n;\n\003GET\022./mlflow/unity-catalog/registered-models/search\032\004\010\002\020\000\020\003*\'(Unity Catalog) Search RegisteredModels'
122
+ _UCMODELREGISTRYSERVICE.methods_by_name['CreateModelVersion']._options = None
123
+ _UCMODELREGISTRYSERVICE.methods_by_name['CreateModelVersion']._serialized_options = b'\362\206\031b\n9\n\004POST\022+/mlflow/unity-catalog/model-versions/create\032\004\010\002\020\000\020\003*#(Unity Catalog) Create ModelVersion'
124
+ _UCMODELREGISTRYSERVICE.methods_by_name['UpdateModelVersion']._options = None
125
+ _UCMODELREGISTRYSERVICE.methods_by_name['UpdateModelVersion']._serialized_options = b'\362\206\031c\n:\n\005PATCH\022+/mlflow/unity-catalog/model-versions/update\032\004\010\002\020\000\020\003*#(Unity Catalog) Update ModelVersion'
126
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteModelVersion']._options = None
127
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteModelVersion']._serialized_options = b'\362\206\031d\n;\n\006DELETE\022+/mlflow/unity-catalog/model-versions/delete\032\004\010\002\020\000\020\003*#(Unity Catalog) Delete ModelVersion'
128
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetModelVersion']._options = None
129
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetModelVersion']._serialized_options = b'\362\206\031[\n5\n\003GET\022(/mlflow/unity-catalog/model-versions/get\032\004\010\002\020\000\020\003* (Unity Catalog) Get ModelVersion'
130
+ _UCMODELREGISTRYSERVICE.methods_by_name['SearchModelVersions']._options = None
131
+ _UCMODELREGISTRYSERVICE.methods_by_name['SearchModelVersions']._serialized_options = b'\362\206\031b\n8\n\003GET\022+/mlflow/unity-catalog/model-versions/search\032\004\010\002\020\000\020\003*$(Unity Catalog) Search ModelVersions'
132
+ _UCMODELREGISTRYSERVICE.methods_by_name['GenerateTemporaryModelVersionCredentials']._options = None
133
+ _UCMODELREGISTRYSERVICE.methods_by_name['GenerateTemporaryModelVersionCredentials']._serialized_options = b'\362\206\031\223\001\nQ\n\004POST\022C/mlflow/unity-catalog/model-versions/generate-temporary-credentials\032\004\010\002\020\000\020\003*<(Unity Catalog) Generate Temporary Model Version Credentials'
134
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetModelVersionDownloadUri']._options = None
135
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetModelVersionDownloadUri']._serialized_options = b'\362\206\031\203\001\nB\n\003GET\0225/mlflow/unity-catalog/model-versions/get-download-uri\032\004\010\002\020\000\020\003*;(Unity Catalog) Get Download URI For ModelVersion Artifacts'
136
+ _UCMODELREGISTRYSERVICE.methods_by_name['FinalizeModelVersion']._options = None
137
+ _UCMODELREGISTRYSERVICE.methods_by_name['FinalizeModelVersion']._serialized_options = b'\362\206\031g\n;\n\004POST\022-/mlflow/unity-catalog/model-versions/finalize\032\004\010\002\020\000\020\003*&(Unity Catalog) Finalize Model Version'
138
+ _UCMODELREGISTRYSERVICE.methods_by_name['SetRegisteredModelAlias']._options = None
139
+ _UCMODELREGISTRYSERVICE.methods_by_name['SetRegisteredModelAlias']._serialized_options = b'\362\206\031j\n;\n\004POST\022-/mlflow/unity-catalog/registered-models/alias\032\004\010\002\020\000\020\003*)(Unity Catalog) Set RegisteredModel Alias'
140
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteRegisteredModelAlias']._options = None
141
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteRegisteredModelAlias']._serialized_options = b'\362\206\031o\n=\n\006DELETE\022-/mlflow/unity-catalog/registered-models/alias\032\004\010\002\020\000\020\003*,(Unity Catalog) Delete RegisteredModel Alias'
142
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetModelVersionByAlias']._options = None
143
+ _UCMODELREGISTRYSERVICE.methods_by_name['GetModelVersionByAlias']._serialized_options = b'\362\206\031i\n:\n\003GET\022-/mlflow/unity-catalog/registered-models/alias\032\004\010\002\020\000\020\003*)(Unity Catalog) Get ModelVersion By Alias'
144
+ _UCMODELREGISTRYSERVICE.methods_by_name['SetRegisteredModelTag']._options = None
145
+ _UCMODELREGISTRYSERVICE.methods_by_name['SetRegisteredModelTag']._serialized_options = b'\362\206\031j\n=\n\004POST\022//mlflow/unity-catalog/registered-models/set-tag\032\004\010\002\020\000\020\003*\'(Unity Catalog) Set RegisteredModel Tag'
146
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteRegisteredModelTag']._options = None
147
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteRegisteredModelTag']._serialized_options = b'\362\206\031r\nB\n\006DELETE\0222/mlflow/unity-catalog/registered-models/delete-tag\032\004\010\002\020\000\020\003**(Unity Catalog) Delete RegisteredModel Tag'
148
+ _UCMODELREGISTRYSERVICE.methods_by_name['SetModelVersionTag']._options = None
149
+ _UCMODELREGISTRYSERVICE.methods_by_name['SetModelVersionTag']._serialized_options = b'\362\206\031d\n:\n\004POST\022,/mlflow/unity-catalog/model-versions/set-tag\032\004\010\002\020\000\020\003*$(Unity Catalog) Set ModelVersion Tag'
150
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteModelVersionTag']._options = None
151
+ _UCMODELREGISTRYSERVICE.methods_by_name['DeleteModelVersionTag']._serialized_options = b'\362\206\031l\n?\n\006DELETE\022//mlflow/unity-catalog/model-versions/delete-tag\032\004\010\002\020\000\020\003*\'(Unity Catalog) Delete ModelVersion Tag'
152
+ _UCMODELREGISTRYSERVICE.methods_by_name['EmitModelVersionLineage']._options = None
153
+ _UCMODELREGISTRYSERVICE.methods_by_name['EmitModelVersionLineage']._serialized_options = b'\362\206\031n\n?\n\004POST\0221/mlflow/unity-catalog/model-versions/emit-lineage\032\004\010\002\020\000\020\003*)(Unity Catalog) Emit ModelVersion Lineage'
154
+ _UCMODELREGISTRYSERVICE.methods_by_name['IsDatabricksSdkModelsArtifactRepositoryEnabled']._options = None
155
+ _UCMODELREGISTRYSERVICE.methods_by_name['IsDatabricksSdkModelsArtifactRepositoryEnabled']._serialized_options = b'\362\206\031\314\001\ni\n\003GET\022\\/mlflow/unity-catalog/registered-models:is-databricks-sdk-models-artifact-repository-enabled\032\004\010\002\020\000\020\003*](Unity Catalog) Use DatabricksSdkModelsArtifactRepository in mlflow client for model registry'
156
+ _UCMODELREGISTRYSERVICE._serialized_start=145
157
+ _UCMODELREGISTRYSERVICE._serialized_end=5841
158
+ UcModelRegistryService = service_reflection.GeneratedServiceType('UcModelRegistryService', (_service.Service,), dict(
159
+ DESCRIPTOR = _UCMODELREGISTRYSERVICE,
160
+ __module__ = 'databricks_uc_registry_service_pb2'
161
+ ))
162
+
163
+ UcModelRegistryService_Stub = service_reflection.GeneratedServiceStubType('UcModelRegistryService_Stub', (UcModelRegistryService,), dict(
164
+ DESCRIPTOR = _UCMODELREGISTRYSERVICE,
165
+ __module__ = 'databricks_uc_registry_service_pb2'
166
+ ))
167
+
168
+
169
+ # @@protoc_insertion_point(module_scope)
170
+
mlflow/protos/facet_feature_statistics_pb2.py ADDED
@@ -0,0 +1,296 @@
1
+
2
+ import google.protobuf
3
+ from packaging.version import Version
4
+ if Version(google.protobuf.__version__).major >= 5:
5
+ # -*- coding: utf-8 -*-
6
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
7
+ # source: facet_feature_statistics.proto
8
+ # Protobuf Python Version: 5.26.0
9
+ """Generated protocol buffer code."""
10
+ from google.protobuf import descriptor as _descriptor
11
+ from google.protobuf import descriptor_pool as _descriptor_pool
12
+ from google.protobuf import symbol_database as _symbol_database
13
+ from google.protobuf.internal import builder as _builder
14
+ # @@protoc_insertion_point(imports)
15
+
16
+ _sym_db = _symbol_database.Default()
17
+
18
+
19
+
20
+
21
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1e\x66\x61\x63\x65t_feature_statistics.proto\x12\x16\x66\x61\x63\x65tFeatureStatistics\"b\n\x1c\x44\x61tasetFeatureStatisticsList\x12\x42\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32\x30.facetFeatureStatistics.DatasetFeatureStatistics\"\x14\n\x04Path\x12\x0c\n\x04step\x18\x01 \x03(\t\"\x9e\x01\n\x18\x44\x61tasetFeatureStatistics\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cnum_examples\x18\x02 \x01(\x03\x12\x1d\n\x15weighted_num_examples\x18\x04 \x01(\x01\x12?\n\x08\x66\x65\x61tures\x18\x03 \x03(\x0b\x32-.facetFeatureStatistics.FeatureNameStatistics\"\xae\x04\n\x15\x46\x65\x61tureNameStatistics\x12\x0e\n\x04name\x18\x01 \x01(\tH\x00\x12,\n\x04path\x18\x08 \x01(\x0b\x32\x1c.facetFeatureStatistics.PathH\x00\x12@\n\x04type\x18\x02 \x01(\x0e\x32\x32.facetFeatureStatistics.FeatureNameStatistics.Type\x12>\n\tnum_stats\x18\x03 \x01(\x0b\x32).facetFeatureStatistics.NumericStatisticsH\x01\x12@\n\x0cstring_stats\x18\x04 \x01(\x0b\x32(.facetFeatureStatistics.StringStatisticsH\x01\x12>\n\x0b\x62ytes_stats\x18\x05 \x01(\x0b\x32\'.facetFeatureStatistics.BytesStatisticsH\x01\x12@\n\x0cstruct_stats\x18\x07 \x01(\x0b\x32(.facetFeatureStatistics.StructStatisticsH\x01\x12=\n\x0c\x63ustom_stats\x18\x06 \x03(\x0b\x32\'.facetFeatureStatistics.CustomStatistic\"=\n\x04Type\x12\x07\n\x03INT\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\n\n\x06STRING\x10\x02\x12\t\n\x05\x42YTES\x10\x03\x12\n\n\x06STRUCT\x10\x04\x42\n\n\x08\x66ield_idB\x07\n\x05stats\"x\n\x18WeightedCommonStatistics\x12\x17\n\x0fnum_non_missing\x18\x01 \x01(\x01\x12\x13\n\x0bnum_missing\x18\x02 \x01(\x01\x12\x16\n\x0e\x61vg_num_values\x18\x03 \x01(\x01\x12\x16\n\x0etot_num_values\x18\x04 \x01(\x01\"\xc3\x01\n\x0f\x43ustomStatistic\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x03num\x18\x02 \x01(\x01H\x00\x12\r\n\x03str\x18\x03 \x01(\tH\x00\x12\x36\n\thistogram\x18\x04 \x01(\x0b\x32!.facetFeatureStatistics.HistogramH\x00\x12?\n\x0erank_histogram\x18\x05 \x01(\x0b\x32%.facetFeatureStatistics.RankHistogramH\x00\x42\x05\n\x03valJ\x04\x08\x06\x10\x07\"\xb9\x02\n\x11NumericStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\x12\x0c\n\x04mean\x18\x02 \x01(\x01\x12\x0f\n\x07std_dev\x18\x03 \x01(\x01\x12\x11\n\tnum_zeros\x18\x04 \x01(\x03\x12\x0b\n\x03min\x18\x05 \x01(\x01\x12\x0e\n\x06median\x18\x06 \x01(\x01\x12\x0b\n\x03max\x18\x07 \x01(\x01\x12\x35\n\nhistograms\x18\x08 \x03(\x0b\x32!.facetFeatureStatistics.Histogram\x12Q\n\x16weighted_numeric_stats\x18\t \x01(\x0b\x32\x31.facetFeatureStatistics.WeightedNumericStatistics\"\xa0\x03\n\x10StringStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\x12\x0e\n\x06unique\x18\x02 \x01(\x03\x12I\n\ntop_values\x18\x03 \x03(\x0b\x32\x35.facetFeatureStatistics.StringStatistics.FreqAndValue\x12\x12\n\navg_length\x18\x04 \x01(\x02\x12=\n\x0erank_histogram\x18\x05 \x01(\x0b\x32%.facetFeatureStatistics.RankHistogram\x12O\n\x15weighted_string_stats\x18\x06 \x01(\x0b\x32\x30.facetFeatureStatistics.WeightedStringStatistics\x1aM\n\x0c\x46reqAndValue\x12\x1b\n\x0f\x64\x65precated_freq\x18\x01 \x01(\x03\x42\x02\x18\x01\x12\r\n\x05value\x18\x02 \x01(\t\x12\x11\n\tfrequency\x18\x03 \x01(\x01\"\x81\x01\n\x19WeightedNumericStatistics\x12\x0c\n\x04mean\x18\x01 \x01(\x01\x12\x0f\n\x07std_dev\x18\x02 \x01(\x01\x12\x0e\n\x06median\x18\x03 \x01(\x01\x12\x35\n\nhistograms\x18\x04 \x03(\x0b\x32!.facetFeatureStatistics.Histogram\"\xa4\x01\n\x18WeightedStringStatistics\x12I\n\ntop_values\x18\x01 \x03(\x0b\x32\x35.facetFeatureStatistics.StringStatistics.FreqAndValue\x12=\n\x0erank_histogram\x18\x02 \x01(\x0b\x32%.facetFeatureStatistics.RankHistogram\"\xac\x01\n\x0f\x42ytesStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\x12\x0e\n\x06unique\x18\x02 \x01(\x03\x12\x15\n\ravg_num_bytes\x18\x03 \x01(\x02\x12\x15\n\rmin_num_bytes\x18\x04 \x01(\x02\x12\x15\n\rmax_num_bytes\x18\x05 \x01(\x02J\x04\x08\x06\x10\x07\"R\n\x10StructStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\"\x88\x03\n\x10\x43ommonStatistics\x12\x17\n\x0fnum_non_missing\x18\x01 \x01(\x03\x12\x13\n\x0bnum_missing\x18\x02 \x01(\x03\x12\x16\n\x0emin_num_values\x18\x03 \x01(\x03\x12\x16\n\x0emax_num_values\x18\x04 \x01(\x03\x12\x16\n\x0e\x61vg_num_values\x18\x05 \x01(\x02\x12\x16\n\x0etot_num_values\x18\x08 \x01(\x03\x12?\n\x14num_values_histogram\x18\x06 \x01(\x0b\x32!.facetFeatureStatistics.Histogram\x12O\n\x15weighted_common_stats\x18\x07 \x01(\x0b\x32\x30.facetFeatureStatistics.WeightedCommonStatistics\x12H\n\x1d\x66\x65\x61ture_list_length_histogram\x18\t \x01(\x0b\x32!.facetFeatureStatistics.HistogramJ\x04\x08\n\x10\x0bJ\x04\x08\x0b\x10\x0c\"\xce\x02\n\tHistogram\x12\x0f\n\x07num_nan\x18\x01 \x01(\x03\x12\x15\n\rnum_undefined\x18\x02 \x01(\x03\x12\x39\n\x07\x62uckets\x18\x03 \x03(\x0b\x32(.facetFeatureStatistics.Histogram.Bucket\x12=\n\x04type\x18\x04 \x01(\x0e\x32/.facetFeatureStatistics.Histogram.HistogramType\x12\x0c\n\x04name\x18\x05 \x01(\t\x1a\x63\n\x06\x42ucket\x12\x11\n\tlow_value\x18\x01 \x01(\x01\x12\x12\n\nhigh_value\x18\x02 \x01(\x01\x12\x1c\n\x10\x64\x65precated_count\x18\x03 \x01(\x03\x42\x02\x18\x01\x12\x14\n\x0csample_count\x18\x04 \x01(\x01\",\n\rHistogramType\x12\x0c\n\x08STANDARD\x10\x00\x12\r\n\tQUANTILES\x10\x01\"\xce\x01\n\rRankHistogram\x12=\n\x07\x62uckets\x18\x01 \x03(\x0b\x32,.facetFeatureStatistics.RankHistogram.Bucket\x12\x0c\n\x04name\x18\x02 \x01(\t\x1ap\n\x06\x42ucket\x12\x10\n\x08low_rank\x18\x01 \x01(\x03\x12\x11\n\thigh_rank\x18\x02 \x01(\x03\x12\x1c\n\x10\x64\x65precated_count\x18\x03 \x01(\x03\x42\x02\x18\x01\x12\r\n\x05label\x18\x04 \x01(\t\x12\x14\n\x0csample_count\x18\x05 \x01(\x01')
22
+
23
+ _globals = globals()
24
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
25
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'facet_feature_statistics_pb2', _globals)
26
+ if not _descriptor._USE_C_DESCRIPTORS:
27
+ DESCRIPTOR._loaded_options = None
28
+ _globals['_STRINGSTATISTICS_FREQANDVALUE'].fields_by_name['deprecated_freq']._loaded_options = None
29
+ _globals['_STRINGSTATISTICS_FREQANDVALUE'].fields_by_name['deprecated_freq']._serialized_options = b'\030\001'
30
+ _globals['_HISTOGRAM_BUCKET'].fields_by_name['deprecated_count']._loaded_options = None
31
+ _globals['_HISTOGRAM_BUCKET'].fields_by_name['deprecated_count']._serialized_options = b'\030\001'
32
+ _globals['_RANKHISTOGRAM_BUCKET'].fields_by_name['deprecated_count']._loaded_options = None
33
+ _globals['_RANKHISTOGRAM_BUCKET'].fields_by_name['deprecated_count']._serialized_options = b'\030\001'
34
+ _globals['_DATASETFEATURESTATISTICSLIST']._serialized_start=58
35
+ _globals['_DATASETFEATURESTATISTICSLIST']._serialized_end=156
36
+ _globals['_PATH']._serialized_start=158
37
+ _globals['_PATH']._serialized_end=178
38
+ _globals['_DATASETFEATURESTATISTICS']._serialized_start=181
39
+ _globals['_DATASETFEATURESTATISTICS']._serialized_end=339
40
+ _globals['_FEATURENAMESTATISTICS']._serialized_start=342
41
+ _globals['_FEATURENAMESTATISTICS']._serialized_end=900
42
+ _globals['_FEATURENAMESTATISTICS_TYPE']._serialized_start=818
43
+ _globals['_FEATURENAMESTATISTICS_TYPE']._serialized_end=879
44
+ _globals['_WEIGHTEDCOMMONSTATISTICS']._serialized_start=902
45
+ _globals['_WEIGHTEDCOMMONSTATISTICS']._serialized_end=1022
46
+ _globals['_CUSTOMSTATISTIC']._serialized_start=1025
47
+ _globals['_CUSTOMSTATISTIC']._serialized_end=1220
48
+ _globals['_NUMERICSTATISTICS']._serialized_start=1223
49
+ _globals['_NUMERICSTATISTICS']._serialized_end=1536
50
+ _globals['_STRINGSTATISTICS']._serialized_start=1539
51
+ _globals['_STRINGSTATISTICS']._serialized_end=1955
52
+ _globals['_STRINGSTATISTICS_FREQANDVALUE']._serialized_start=1878
53
+ _globals['_STRINGSTATISTICS_FREQANDVALUE']._serialized_end=1955
54
+ _globals['_WEIGHTEDNUMERICSTATISTICS']._serialized_start=1958
55
+ _globals['_WEIGHTEDNUMERICSTATISTICS']._serialized_end=2087
56
+ _globals['_WEIGHTEDSTRINGSTATISTICS']._serialized_start=2090
57
+ _globals['_WEIGHTEDSTRINGSTATISTICS']._serialized_end=2254
58
+ _globals['_BYTESSTATISTICS']._serialized_start=2257
59
+ _globals['_BYTESSTATISTICS']._serialized_end=2429
60
+ _globals['_STRUCTSTATISTICS']._serialized_start=2431
61
+ _globals['_STRUCTSTATISTICS']._serialized_end=2513
62
+ _globals['_COMMONSTATISTICS']._serialized_start=2516
63
+ _globals['_COMMONSTATISTICS']._serialized_end=2908
64
+ _globals['_HISTOGRAM']._serialized_start=2911
65
+ _globals['_HISTOGRAM']._serialized_end=3245
66
+ _globals['_HISTOGRAM_BUCKET']._serialized_start=3100
67
+ _globals['_HISTOGRAM_BUCKET']._serialized_end=3199
68
+ _globals['_HISTOGRAM_HISTOGRAMTYPE']._serialized_start=3201
69
+ _globals['_HISTOGRAM_HISTOGRAMTYPE']._serialized_end=3245
70
+ _globals['_RANKHISTOGRAM']._serialized_start=3248
71
+ _globals['_RANKHISTOGRAM']._serialized_end=3454
72
+ _globals['_RANKHISTOGRAM_BUCKET']._serialized_start=3342
73
+ _globals['_RANKHISTOGRAM_BUCKET']._serialized_end=3454
74
+ # @@protoc_insertion_point(module_scope)
75
+
76
+ else:
77
+ # -*- coding: utf-8 -*-
78
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
79
+ # source: facet_feature_statistics.proto
80
+ """Generated protocol buffer code."""
81
+ from google.protobuf import descriptor as _descriptor
82
+ from google.protobuf import descriptor_pool as _descriptor_pool
83
+ from google.protobuf import message as _message
84
+ from google.protobuf import reflection as _reflection
85
+ from google.protobuf import symbol_database as _symbol_database
86
+ # @@protoc_insertion_point(imports)
87
+
88
+ _sym_db = _symbol_database.Default()
89
+
90
+
91
+
92
+
93
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1e\x66\x61\x63\x65t_feature_statistics.proto\x12\x16\x66\x61\x63\x65tFeatureStatistics\"b\n\x1c\x44\x61tasetFeatureStatisticsList\x12\x42\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32\x30.facetFeatureStatistics.DatasetFeatureStatistics\"\x14\n\x04Path\x12\x0c\n\x04step\x18\x01 \x03(\t\"\x9e\x01\n\x18\x44\x61tasetFeatureStatistics\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cnum_examples\x18\x02 \x01(\x03\x12\x1d\n\x15weighted_num_examples\x18\x04 \x01(\x01\x12?\n\x08\x66\x65\x61tures\x18\x03 \x03(\x0b\x32-.facetFeatureStatistics.FeatureNameStatistics\"\xae\x04\n\x15\x46\x65\x61tureNameStatistics\x12\x0e\n\x04name\x18\x01 \x01(\tH\x00\x12,\n\x04path\x18\x08 \x01(\x0b\x32\x1c.facetFeatureStatistics.PathH\x00\x12@\n\x04type\x18\x02 \x01(\x0e\x32\x32.facetFeatureStatistics.FeatureNameStatistics.Type\x12>\n\tnum_stats\x18\x03 \x01(\x0b\x32).facetFeatureStatistics.NumericStatisticsH\x01\x12@\n\x0cstring_stats\x18\x04 \x01(\x0b\x32(.facetFeatureStatistics.StringStatisticsH\x01\x12>\n\x0b\x62ytes_stats\x18\x05 \x01(\x0b\x32\'.facetFeatureStatistics.BytesStatisticsH\x01\x12@\n\x0cstruct_stats\x18\x07 \x01(\x0b\x32(.facetFeatureStatistics.StructStatisticsH\x01\x12=\n\x0c\x63ustom_stats\x18\x06 \x03(\x0b\x32\'.facetFeatureStatistics.CustomStatistic\"=\n\x04Type\x12\x07\n\x03INT\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\n\n\x06STRING\x10\x02\x12\t\n\x05\x42YTES\x10\x03\x12\n\n\x06STRUCT\x10\x04\x42\n\n\x08\x66ield_idB\x07\n\x05stats\"x\n\x18WeightedCommonStatistics\x12\x17\n\x0fnum_non_missing\x18\x01 \x01(\x01\x12\x13\n\x0bnum_missing\x18\x02 \x01(\x01\x12\x16\n\x0e\x61vg_num_values\x18\x03 \x01(\x01\x12\x16\n\x0etot_num_values\x18\x04 \x01(\x01\"\xc3\x01\n\x0f\x43ustomStatistic\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x03num\x18\x02 \x01(\x01H\x00\x12\r\n\x03str\x18\x03 \x01(\tH\x00\x12\x36\n\thistogram\x18\x04 \x01(\x0b\x32!.facetFeatureStatistics.HistogramH\x00\x12?\n\x0erank_histogram\x18\x05 \x01(\x0b\x32%.facetFeatureStatistics.RankHistogramH\x00\x42\x05\n\x03valJ\x04\x08\x06\x10\x07\"\xb9\x02\n\x11NumericStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\x12\x0c\n\x04mean\x18\x02 \x01(\x01\x12\x0f\n\x07std_dev\x18\x03 \x01(\x01\x12\x11\n\tnum_zeros\x18\x04 \x01(\x03\x12\x0b\n\x03min\x18\x05 \x01(\x01\x12\x0e\n\x06median\x18\x06 \x01(\x01\x12\x0b\n\x03max\x18\x07 \x01(\x01\x12\x35\n\nhistograms\x18\x08 \x03(\x0b\x32!.facetFeatureStatistics.Histogram\x12Q\n\x16weighted_numeric_stats\x18\t \x01(\x0b\x32\x31.facetFeatureStatistics.WeightedNumericStatistics\"\xa0\x03\n\x10StringStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\x12\x0e\n\x06unique\x18\x02 \x01(\x03\x12I\n\ntop_values\x18\x03 \x03(\x0b\x32\x35.facetFeatureStatistics.StringStatistics.FreqAndValue\x12\x12\n\navg_length\x18\x04 \x01(\x02\x12=\n\x0erank_histogram\x18\x05 \x01(\x0b\x32%.facetFeatureStatistics.RankHistogram\x12O\n\x15weighted_string_stats\x18\x06 \x01(\x0b\x32\x30.facetFeatureStatistics.WeightedStringStatistics\x1aM\n\x0c\x46reqAndValue\x12\x1b\n\x0f\x64\x65precated_freq\x18\x01 \x01(\x03\x42\x02\x18\x01\x12\r\n\x05value\x18\x02 \x01(\t\x12\x11\n\tfrequency\x18\x03 \x01(\x01\"\x81\x01\n\x19WeightedNumericStatistics\x12\x0c\n\x04mean\x18\x01 \x01(\x01\x12\x0f\n\x07std_dev\x18\x02 \x01(\x01\x12\x0e\n\x06median\x18\x03 \x01(\x01\x12\x35\n\nhistograms\x18\x04 \x03(\x0b\x32!.facetFeatureStatistics.Histogram\"\xa4\x01\n\x18WeightedStringStatistics\x12I\n\ntop_values\x18\x01 \x03(\x0b\x32\x35.facetFeatureStatistics.StringStatistics.FreqAndValue\x12=\n\x0erank_histogram\x18\x02 \x01(\x0b\x32%.facetFeatureStatistics.RankHistogram\"\xac\x01\n\x0f\x42ytesStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\x12\x0e\n\x06unique\x18\x02 \x01(\x03\x12\x15\n\ravg_num_bytes\x18\x03 \x01(\x02\x12\x15\n\rmin_num_bytes\x18\x04 \x01(\x02\x12\x15\n\rmax_num_bytes\x18\x05 \x01(\x02J\x04\x08\x06\x10\x07\"R\n\x10StructStatistics\x12>\n\x0c\x63ommon_stats\x18\x01 \x01(\x0b\x32(.facetFeatureStatistics.CommonStatistics\"\x88\x03\n\x10\x43ommonStatistics\x12\x17\n\x0fnum_non_missing\x18\x01 \x01(\x03\x12\x13\n\x0bnum_missing\x18\x02 \x01(\x03\x12\x16\n\x0emin_num_values\x18\x03 \x01(\x03\x12\x16\n\x0emax_num_values\x18\x04 \x01(\x03\x12\x16\n\x0e\x61vg_num_values\x18\x05 \x01(\x02\x12\x16\n\x0etot_num_values\x18\x08 \x01(\x03\x12?\n\x14num_values_histogram\x18\x06 \x01(\x0b\x32!.facetFeatureStatistics.Histogram\x12O\n\x15weighted_common_stats\x18\x07 \x01(\x0b\x32\x30.facetFeatureStatistics.WeightedCommonStatistics\x12H\n\x1d\x66\x65\x61ture_list_length_histogram\x18\t \x01(\x0b\x32!.facetFeatureStatistics.HistogramJ\x04\x08\n\x10\x0bJ\x04\x08\x0b\x10\x0c\"\xce\x02\n\tHistogram\x12\x0f\n\x07num_nan\x18\x01 \x01(\x03\x12\x15\n\rnum_undefined\x18\x02 \x01(\x03\x12\x39\n\x07\x62uckets\x18\x03 \x03(\x0b\x32(.facetFeatureStatistics.Histogram.Bucket\x12=\n\x04type\x18\x04 \x01(\x0e\x32/.facetFeatureStatistics.Histogram.HistogramType\x12\x0c\n\x04name\x18\x05 \x01(\t\x1a\x63\n\x06\x42ucket\x12\x11\n\tlow_value\x18\x01 \x01(\x01\x12\x12\n\nhigh_value\x18\x02 \x01(\x01\x12\x1c\n\x10\x64\x65precated_count\x18\x03 \x01(\x03\x42\x02\x18\x01\x12\x14\n\x0csample_count\x18\x04 \x01(\x01\",\n\rHistogramType\x12\x0c\n\x08STANDARD\x10\x00\x12\r\n\tQUANTILES\x10\x01\"\xce\x01\n\rRankHistogram\x12=\n\x07\x62uckets\x18\x01 \x03(\x0b\x32,.facetFeatureStatistics.RankHistogram.Bucket\x12\x0c\n\x04name\x18\x02 \x01(\t\x1ap\n\x06\x42ucket\x12\x10\n\x08low_rank\x18\x01 \x01(\x03\x12\x11\n\thigh_rank\x18\x02 \x01(\x03\x12\x1c\n\x10\x64\x65precated_count\x18\x03 \x01(\x03\x42\x02\x18\x01\x12\r\n\x05label\x18\x04 \x01(\t\x12\x14\n\x0csample_count\x18\x05 \x01(\x01')
94
+
95
+
96
+
97
+ _DATASETFEATURESTATISTICSLIST = DESCRIPTOR.message_types_by_name['DatasetFeatureStatisticsList']
98
+ _PATH = DESCRIPTOR.message_types_by_name['Path']
99
+ _DATASETFEATURESTATISTICS = DESCRIPTOR.message_types_by_name['DatasetFeatureStatistics']
100
+ _FEATURENAMESTATISTICS = DESCRIPTOR.message_types_by_name['FeatureNameStatistics']
101
+ _WEIGHTEDCOMMONSTATISTICS = DESCRIPTOR.message_types_by_name['WeightedCommonStatistics']
102
+ _CUSTOMSTATISTIC = DESCRIPTOR.message_types_by_name['CustomStatistic']
103
+ _NUMERICSTATISTICS = DESCRIPTOR.message_types_by_name['NumericStatistics']
104
+ _STRINGSTATISTICS = DESCRIPTOR.message_types_by_name['StringStatistics']
105
+ _STRINGSTATISTICS_FREQANDVALUE = _STRINGSTATISTICS.nested_types_by_name['FreqAndValue']
106
+ _WEIGHTEDNUMERICSTATISTICS = DESCRIPTOR.message_types_by_name['WeightedNumericStatistics']
107
+ _WEIGHTEDSTRINGSTATISTICS = DESCRIPTOR.message_types_by_name['WeightedStringStatistics']
108
+ _BYTESSTATISTICS = DESCRIPTOR.message_types_by_name['BytesStatistics']
109
+ _STRUCTSTATISTICS = DESCRIPTOR.message_types_by_name['StructStatistics']
110
+ _COMMONSTATISTICS = DESCRIPTOR.message_types_by_name['CommonStatistics']
111
+ _HISTOGRAM = DESCRIPTOR.message_types_by_name['Histogram']
112
+ _HISTOGRAM_BUCKET = _HISTOGRAM.nested_types_by_name['Bucket']
113
+ _RANKHISTOGRAM = DESCRIPTOR.message_types_by_name['RankHistogram']
114
+ _RANKHISTOGRAM_BUCKET = _RANKHISTOGRAM.nested_types_by_name['Bucket']
115
+ _FEATURENAMESTATISTICS_TYPE = _FEATURENAMESTATISTICS.enum_types_by_name['Type']
116
+ _HISTOGRAM_HISTOGRAMTYPE = _HISTOGRAM.enum_types_by_name['HistogramType']
117
+ DatasetFeatureStatisticsList = _reflection.GeneratedProtocolMessageType('DatasetFeatureStatisticsList', (_message.Message,), {
118
+ 'DESCRIPTOR' : _DATASETFEATURESTATISTICSLIST,
119
+ '__module__' : 'facet_feature_statistics_pb2'
120
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.DatasetFeatureStatisticsList)
121
+ })
122
+ _sym_db.RegisterMessage(DatasetFeatureStatisticsList)
123
+
124
+ Path = _reflection.GeneratedProtocolMessageType('Path', (_message.Message,), {
125
+ 'DESCRIPTOR' : _PATH,
126
+ '__module__' : 'facet_feature_statistics_pb2'
127
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.Path)
128
+ })
129
+ _sym_db.RegisterMessage(Path)
130
+
131
+ DatasetFeatureStatistics = _reflection.GeneratedProtocolMessageType('DatasetFeatureStatistics', (_message.Message,), {
132
+ 'DESCRIPTOR' : _DATASETFEATURESTATISTICS,
133
+ '__module__' : 'facet_feature_statistics_pb2'
134
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.DatasetFeatureStatistics)
135
+ })
136
+ _sym_db.RegisterMessage(DatasetFeatureStatistics)
137
+
138
+ FeatureNameStatistics = _reflection.GeneratedProtocolMessageType('FeatureNameStatistics', (_message.Message,), {
139
+ 'DESCRIPTOR' : _FEATURENAMESTATISTICS,
140
+ '__module__' : 'facet_feature_statistics_pb2'
141
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.FeatureNameStatistics)
142
+ })
143
+ _sym_db.RegisterMessage(FeatureNameStatistics)
144
+
145
+ WeightedCommonStatistics = _reflection.GeneratedProtocolMessageType('WeightedCommonStatistics', (_message.Message,), {
146
+ 'DESCRIPTOR' : _WEIGHTEDCOMMONSTATISTICS,
147
+ '__module__' : 'facet_feature_statistics_pb2'
148
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.WeightedCommonStatistics)
149
+ })
150
+ _sym_db.RegisterMessage(WeightedCommonStatistics)
151
+
152
+ CustomStatistic = _reflection.GeneratedProtocolMessageType('CustomStatistic', (_message.Message,), {
153
+ 'DESCRIPTOR' : _CUSTOMSTATISTIC,
154
+ '__module__' : 'facet_feature_statistics_pb2'
155
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.CustomStatistic)
156
+ })
157
+ _sym_db.RegisterMessage(CustomStatistic)
158
+
159
+ NumericStatistics = _reflection.GeneratedProtocolMessageType('NumericStatistics', (_message.Message,), {
160
+ 'DESCRIPTOR' : _NUMERICSTATISTICS,
161
+ '__module__' : 'facet_feature_statistics_pb2'
162
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.NumericStatistics)
163
+ })
164
+ _sym_db.RegisterMessage(NumericStatistics)
165
+
166
+ StringStatistics = _reflection.GeneratedProtocolMessageType('StringStatistics', (_message.Message,), {
167
+
168
+ 'FreqAndValue' : _reflection.GeneratedProtocolMessageType('FreqAndValue', (_message.Message,), {
169
+ 'DESCRIPTOR' : _STRINGSTATISTICS_FREQANDVALUE,
170
+ '__module__' : 'facet_feature_statistics_pb2'
171
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.StringStatistics.FreqAndValue)
172
+ })
173
+ ,
174
+ 'DESCRIPTOR' : _STRINGSTATISTICS,
175
+ '__module__' : 'facet_feature_statistics_pb2'
176
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.StringStatistics)
177
+ })
178
+ _sym_db.RegisterMessage(StringStatistics)
179
+ _sym_db.RegisterMessage(StringStatistics.FreqAndValue)
180
+
181
+ WeightedNumericStatistics = _reflection.GeneratedProtocolMessageType('WeightedNumericStatistics', (_message.Message,), {
182
+ 'DESCRIPTOR' : _WEIGHTEDNUMERICSTATISTICS,
183
+ '__module__' : 'facet_feature_statistics_pb2'
184
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.WeightedNumericStatistics)
185
+ })
186
+ _sym_db.RegisterMessage(WeightedNumericStatistics)
187
+
188
+ WeightedStringStatistics = _reflection.GeneratedProtocolMessageType('WeightedStringStatistics', (_message.Message,), {
189
+ 'DESCRIPTOR' : _WEIGHTEDSTRINGSTATISTICS,
190
+ '__module__' : 'facet_feature_statistics_pb2'
191
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.WeightedStringStatistics)
192
+ })
193
+ _sym_db.RegisterMessage(WeightedStringStatistics)
194
+
195
+ BytesStatistics = _reflection.GeneratedProtocolMessageType('BytesStatistics', (_message.Message,), {
196
+ 'DESCRIPTOR' : _BYTESSTATISTICS,
197
+ '__module__' : 'facet_feature_statistics_pb2'
198
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.BytesStatistics)
199
+ })
200
+ _sym_db.RegisterMessage(BytesStatistics)
201
+
202
+ StructStatistics = _reflection.GeneratedProtocolMessageType('StructStatistics', (_message.Message,), {
203
+ 'DESCRIPTOR' : _STRUCTSTATISTICS,
204
+ '__module__' : 'facet_feature_statistics_pb2'
205
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.StructStatistics)
206
+ })
207
+ _sym_db.RegisterMessage(StructStatistics)
208
+
209
+ CommonStatistics = _reflection.GeneratedProtocolMessageType('CommonStatistics', (_message.Message,), {
210
+ 'DESCRIPTOR' : _COMMONSTATISTICS,
211
+ '__module__' : 'facet_feature_statistics_pb2'
212
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.CommonStatistics)
213
+ })
214
+ _sym_db.RegisterMessage(CommonStatistics)
215
+
216
+ Histogram = _reflection.GeneratedProtocolMessageType('Histogram', (_message.Message,), {
217
+
218
+ 'Bucket' : _reflection.GeneratedProtocolMessageType('Bucket', (_message.Message,), {
219
+ 'DESCRIPTOR' : _HISTOGRAM_BUCKET,
220
+ '__module__' : 'facet_feature_statistics_pb2'
221
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.Histogram.Bucket)
222
+ })
223
+ ,
224
+ 'DESCRIPTOR' : _HISTOGRAM,
225
+ '__module__' : 'facet_feature_statistics_pb2'
226
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.Histogram)
227
+ })
228
+ _sym_db.RegisterMessage(Histogram)
229
+ _sym_db.RegisterMessage(Histogram.Bucket)
230
+
231
+ RankHistogram = _reflection.GeneratedProtocolMessageType('RankHistogram', (_message.Message,), {
232
+
233
+ 'Bucket' : _reflection.GeneratedProtocolMessageType('Bucket', (_message.Message,), {
234
+ 'DESCRIPTOR' : _RANKHISTOGRAM_BUCKET,
235
+ '__module__' : 'facet_feature_statistics_pb2'
236
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.RankHistogram.Bucket)
237
+ })
238
+ ,
239
+ 'DESCRIPTOR' : _RANKHISTOGRAM,
240
+ '__module__' : 'facet_feature_statistics_pb2'
241
+ # @@protoc_insertion_point(class_scope:facetFeatureStatistics.RankHistogram)
242
+ })
243
+ _sym_db.RegisterMessage(RankHistogram)
244
+ _sym_db.RegisterMessage(RankHistogram.Bucket)
245
+
246
+ if _descriptor._USE_C_DESCRIPTORS == False:
247
+
248
+ DESCRIPTOR._options = None
249
+ _STRINGSTATISTICS_FREQANDVALUE.fields_by_name['deprecated_freq']._options = None
250
+ _STRINGSTATISTICS_FREQANDVALUE.fields_by_name['deprecated_freq']._serialized_options = b'\030\001'
251
+ _HISTOGRAM_BUCKET.fields_by_name['deprecated_count']._options = None
252
+ _HISTOGRAM_BUCKET.fields_by_name['deprecated_count']._serialized_options = b'\030\001'
253
+ _RANKHISTOGRAM_BUCKET.fields_by_name['deprecated_count']._options = None
254
+ _RANKHISTOGRAM_BUCKET.fields_by_name['deprecated_count']._serialized_options = b'\030\001'
255
+ _DATASETFEATURESTATISTICSLIST._serialized_start=58
256
+ _DATASETFEATURESTATISTICSLIST._serialized_end=156
257
+ _PATH._serialized_start=158
258
+ _PATH._serialized_end=178
259
+ _DATASETFEATURESTATISTICS._serialized_start=181
260
+ _DATASETFEATURESTATISTICS._serialized_end=339
261
+ _FEATURENAMESTATISTICS._serialized_start=342
262
+ _FEATURENAMESTATISTICS._serialized_end=900
263
+ _FEATURENAMESTATISTICS_TYPE._serialized_start=818
264
+ _FEATURENAMESTATISTICS_TYPE._serialized_end=879
265
+ _WEIGHTEDCOMMONSTATISTICS._serialized_start=902
266
+ _WEIGHTEDCOMMONSTATISTICS._serialized_end=1022
267
+ _CUSTOMSTATISTIC._serialized_start=1025
268
+ _CUSTOMSTATISTIC._serialized_end=1220
269
+ _NUMERICSTATISTICS._serialized_start=1223
270
+ _NUMERICSTATISTICS._serialized_end=1536
271
+ _STRINGSTATISTICS._serialized_start=1539
272
+ _STRINGSTATISTICS._serialized_end=1955
273
+ _STRINGSTATISTICS_FREQANDVALUE._serialized_start=1878
274
+ _STRINGSTATISTICS_FREQANDVALUE._serialized_end=1955
275
+ _WEIGHTEDNUMERICSTATISTICS._serialized_start=1958
276
+ _WEIGHTEDNUMERICSTATISTICS._serialized_end=2087
277
+ _WEIGHTEDSTRINGSTATISTICS._serialized_start=2090
278
+ _WEIGHTEDSTRINGSTATISTICS._serialized_end=2254
279
+ _BYTESSTATISTICS._serialized_start=2257
280
+ _BYTESSTATISTICS._serialized_end=2429
281
+ _STRUCTSTATISTICS._serialized_start=2431
282
+ _STRUCTSTATISTICS._serialized_end=2513
283
+ _COMMONSTATISTICS._serialized_start=2516
284
+ _COMMONSTATISTICS._serialized_end=2908
285
+ _HISTOGRAM._serialized_start=2911
286
+ _HISTOGRAM._serialized_end=3245
287
+ _HISTOGRAM_BUCKET._serialized_start=3100
288
+ _HISTOGRAM_BUCKET._serialized_end=3199
289
+ _HISTOGRAM_HISTOGRAMTYPE._serialized_start=3201
290
+ _HISTOGRAM_HISTOGRAMTYPE._serialized_end=3245
291
+ _RANKHISTOGRAM._serialized_start=3248
292
+ _RANKHISTOGRAM._serialized_end=3454
293
+ _RANKHISTOGRAM_BUCKET._serialized_start=3342
294
+ _RANKHISTOGRAM_BUCKET._serialized_end=3454
295
+ # @@protoc_insertion_point(module_scope)
296
+
mlflow/protos/internal_pb2.py ADDED
@@ -0,0 +1,77 @@
1
+
2
+ import google.protobuf
3
+ from packaging.version import Version
4
+ if Version(google.protobuf.__version__).major >= 5:
5
+ # -*- coding: utf-8 -*-
6
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
7
+ # source: internal.proto
8
+ # Protobuf Python Version: 5.26.0
9
+ """Generated protocol buffer code."""
10
+ from google.protobuf import descriptor as _descriptor
11
+ from google.protobuf import descriptor_pool as _descriptor_pool
12
+ from google.protobuf import symbol_database as _symbol_database
13
+ from google.protobuf.internal import builder as _builder
14
+ # @@protoc_insertion_point(imports)
15
+
16
+ _sym_db = _symbol_database.Default()
17
+
18
+
19
+ from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
20
+
21
+
22
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0einternal.proto\x12\x0fmlflow.internal\x1a\x15scalapb/scalapb.proto*2\n\x0fInputVertexType\x12\x07\n\x03RUN\x10\x01\x12\x0b\n\x07\x44\x41TASET\x10\x02\x12\t\n\x05MODEL\x10\x03*4\n\x10OutputVertexType\x12\x0e\n\nRUN_OUTPUT\x10\x01\x12\x10\n\x0cMODEL_OUTPUT\x10\x02\x42#\n\x19org.mlflow.internal.proto\x90\x01\x01\xe2?\x02\x10\x01')
23
+
24
+ _globals = globals()
25
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
26
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'internal_pb2', _globals)
27
+ if not _descriptor._USE_C_DESCRIPTORS:
28
+ _globals['DESCRIPTOR']._loaded_options = None
29
+ _globals['DESCRIPTOR']._serialized_options = b'\n\031org.mlflow.internal.proto\220\001\001\342?\002\020\001'
30
+ _globals['_INPUTVERTEXTYPE']._serialized_start=58
31
+ _globals['_INPUTVERTEXTYPE']._serialized_end=108
32
+ _globals['_OUTPUTVERTEXTYPE']._serialized_start=110
33
+ _globals['_OUTPUTVERTEXTYPE']._serialized_end=162
34
+ # @@protoc_insertion_point(module_scope)
35
+
36
+ else:
37
+ # -*- coding: utf-8 -*-
38
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
39
+ # source: internal.proto
40
+ """Generated protocol buffer code."""
41
+ from google.protobuf.internal import enum_type_wrapper
42
+ from google.protobuf import descriptor as _descriptor
43
+ from google.protobuf import descriptor_pool as _descriptor_pool
44
+ from google.protobuf import message as _message
45
+ from google.protobuf import reflection as _reflection
46
+ from google.protobuf import symbol_database as _symbol_database
47
+ # @@protoc_insertion_point(imports)
48
+
49
+ _sym_db = _symbol_database.Default()
50
+
51
+
52
+ from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
53
+
54
+
55
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0einternal.proto\x12\x0fmlflow.internal\x1a\x15scalapb/scalapb.proto*2\n\x0fInputVertexType\x12\x07\n\x03RUN\x10\x01\x12\x0b\n\x07\x44\x41TASET\x10\x02\x12\t\n\x05MODEL\x10\x03*4\n\x10OutputVertexType\x12\x0e\n\nRUN_OUTPUT\x10\x01\x12\x10\n\x0cMODEL_OUTPUT\x10\x02\x42#\n\x19org.mlflow.internal.proto\x90\x01\x01\xe2?\x02\x10\x01')
56
+
57
+ _INPUTVERTEXTYPE = DESCRIPTOR.enum_types_by_name['InputVertexType']
58
+ InputVertexType = enum_type_wrapper.EnumTypeWrapper(_INPUTVERTEXTYPE)
59
+ _OUTPUTVERTEXTYPE = DESCRIPTOR.enum_types_by_name['OutputVertexType']
60
+ OutputVertexType = enum_type_wrapper.EnumTypeWrapper(_OUTPUTVERTEXTYPE)
61
+ RUN = 1
62
+ DATASET = 2
63
+ MODEL = 3
64
+ RUN_OUTPUT = 1
65
+ MODEL_OUTPUT = 2
66
+
67
+
68
+ if _descriptor._USE_C_DESCRIPTORS == False:
69
+
70
+ DESCRIPTOR._options = None
71
+ DESCRIPTOR._serialized_options = b'\n\031org.mlflow.internal.proto\220\001\001\342?\002\020\001'
72
+ _INPUTVERTEXTYPE._serialized_start=58
73
+ _INPUTVERTEXTYPE._serialized_end=108
74
+ _OUTPUTVERTEXTYPE._serialized_start=110
75
+ _OUTPUTVERTEXTYPE._serialized_end=162
76
+ # @@protoc_insertion_point(module_scope)
77
+