datarobot · mjnitz02 · Feb 3, 2025 · Feb 3, 2025 · Feb 3, 2025 · mbromanowsky
diff --git a/datarobot_provider/_experimental/__init__.py b/datarobot_provider/_experimental/__init__.py
@@ -0,0 +1,19 @@
+# Copyright 2025 DataRobot, Inc. and its affiliates.
+#
+# All rights reserved.
+#
+# This is proprietary source code of DataRobot, Inc. and its affiliates.
+#
+# Released under the terms of DataRobot Tool and Utility Agreement.
+import logging
+
+logger = logging.getLogger(__package__)
+
+experimental_warning = (
+    "You have imported from the _experimental directory.\n"
+    "This directory is used for unreleased datarobot features.\n"
+    "Unless you specifically know better,"
+    " you don't have the access to use this functionality in the app, so this code will not work."
+)
-experimental_warning = (
-    "You have imported from the _experimental directory.\n"
-    "This directory is used for unreleased datarobot features.\n"
-    "Unless you specifically know better,"
-    " you don't have the access to use this functionality in the app, so this code will not work."
-)
+experimental_warning = (
+    "You have imported from the _experimental directory.\n"
+    "This directory is used for unreleased datarobot features.\n"
+    "Code from this directory may be untested and may change without warning."
+)
-experimental_warning = (
-    "You have imported from the _experimental directory.\n"
-    "This directory is used for unreleased datarobot features.\n"
-    "Unless you specifically know better,"
-    " you don't have the access to use this functionality in the app, so this code will not work."
-)
+experimental_warning = (
+    "You have imported from the _experimental directory.\n"
+    "This directory is used for unreleased DataRobot features.\n"
+    "Code from this directory may be untested and may change without warning."
+)
-experimental_warning = (
-    "You have imported from the _experimental directory.\n"
-    "This directory is used for unreleased datarobot features.\n"
-    "Unless you specifically know better,"
-    " you don't have the access to use this functionality in the app, so this code will not work."
-)
+experimental_warning = (
+    "You have imported from the _experimental directory.\n"
+    "This directory is used for unreleased datarobot features.\n"
+    "Code from this directory may be untested and may change without warning."
+)
-experimental_warning = (
-    "You have imported from the _experimental directory.\n"
-    "This directory is used for unreleased datarobot features.\n"
-    "Unless you specifically know better,"
-    " you don't have the access to use this functionality in the app, so this code will not work."
-)
+experimental_warning = (
+    "You have imported from the _experimental directory.\n"
+    "This directory is used for unreleased DataRobot features.\n"
+    "Code from this directory may be untested and may change without warning."
+)
+
+logger.warning(experimental_warning)
diff --git a/...vider/example_dags/deprecated/__init__.py → ...er/_experimental/example_dags/__init__.py b/...vider/example_dags/deprecated/__init__.py → ...er/_experimental/example_dags/__init__.py
diff --git a/...d/advanced_datarobot_pipeline_jdbc_dag.py → ...s/advanced_datarobot_pipeline_jdbc_dag.py b/...d/advanced_datarobot_pipeline_jdbc_dag.py → ...s/advanced_datarobot_pipeline_jdbc_dag.py
diff --git a/...tarobot_autopilot_options_pipeline_dag.py → ...tarobot_autopilot_options_pipeline_dag.py b/...tarobot_autopilot_options_pipeline_dag.py → ...tarobot_autopilot_options_pipeline_dag.py
diff --git a/...ted/datarobot_aws_s3_batch_scoring_dag.py → ...ags/datarobot_aws_s3_batch_scoring_dag.py b/...ted/datarobot_aws_s3_batch_scoring_dag.py → ...ags/datarobot_aws_s3_batch_scoring_dag.py
diff --git a/...arobot_azure_storage_batch_scoring_dag.py → ...arobot_azure_storage_batch_scoring_dag.py b/...arobot_azure_storage_batch_scoring_dag.py → ...arobot_azure_storage_batch_scoring_dag.py
diff --git a/...ted/datarobot_batch_monitoring_job_dag.py → ...ags/datarobot_batch_monitoring_job_dag.py b/...ted/datarobot_batch_monitoring_job_dag.py → ...ags/datarobot_batch_monitoring_job_dag.py
diff --git a/.../datarobot_batch_scoring_templated_dag.py → .../datarobot_batch_scoring_templated_dag.py b/.../datarobot_batch_scoring_templated_dag.py → .../datarobot_batch_scoring_templated_dag.py
diff --git a/...d/datarobot_bigquery_batch_scoring_dag.py → ...s/datarobot_bigquery_batch_scoring_dag.py b/...d/datarobot_bigquery_batch_scoring_dag.py → ...s/datarobot_bigquery_batch_scoring_dag.py
diff --git a/...bot_create_project_from_ai_catalog_dag.py → ...bot_create_project_from_ai_catalog_dag.py b/...bot_create_project_from_ai_catalog_dag.py → ...bot_create_project_from_ai_catalog_dag.py
diff --git a/...reate_project_from_dataset_version_dag.py → ...reate_project_from_dataset_version_dag.py b/...reate_project_from_dataset_version_dag.py → ...reate_project_from_dataset_version_dag.py
diff --git a/...ed/datarobot_custom_model_pipeline_dag.py → ...gs/datarobot_custom_model_pipeline_dag.py b/...ed/datarobot_custom_model_pipeline_dag.py → ...gs/datarobot_custom_model_pipeline_dag.py
diff --git a/...robot_custom_partitioning_pipeline_dag.py → ...robot_custom_partitioning_pipeline_dag.py b/...robot_custom_partitioning_pipeline_dag.py → ...robot_custom_partitioning_pipeline_dag.py
diff --git a/...ated/datarobot_dataset_new_version_dag.py → ...dags/datarobot_dataset_new_version_dag.py b/...ated/datarobot_dataset_new_version_dag.py → ...dags/datarobot_dataset_new_version_dag.py
diff --git a/...eprecated/datarobot_dataset_upload_dag.py → ...mple_dags/datarobot_dataset_upload_dag.py b/...eprecated/datarobot_dataset_upload_dag.py → ...mple_dags/datarobot_dataset_upload_dag.py
diff --git a/...bot_datetime_partitioning_pipeline_dag.py → ...bot_datetime_partitioning_pipeline_dag.py b/...bot_datetime_partitioning_pipeline_dag.py → ...bot_datetime_partitioning_pipeline_dag.py
diff --git a/.../datarobot_external_model_pipeline_dag.py → .../datarobot_external_model_pipeline_dag.py b/.../datarobot_external_model_pipeline_dag.py → .../datarobot_external_model_pipeline_dag.py
diff --git a/...tarobot_feature_discovery_pipeline_dag.py → ...tarobot_feature_discovery_pipeline_dag.py b/...tarobot_feature_discovery_pipeline_dag.py → ...tarobot_feature_discovery_pipeline_dag.py
diff --git a/...atarobot_gcp_storage_batch_scoring_dag.py → ...atarobot_gcp_storage_batch_scoring_dag.py b/...atarobot_gcp_storage_batch_scoring_dag.py → ...atarobot_gcp_storage_batch_scoring_dag.py
diff --git a/...deprecated/datarobot_get_datastore_dag.py → ...ample_dags/datarobot_get_datastore_dag.py b/...deprecated/datarobot_get_datastore_dag.py → ...ample_dags/datarobot_get_datastore_dag.py
diff --git a/...cated/datarobot_jdbc_batch_scoring_dag.py → ..._dags/datarobot_jdbc_batch_scoring_dag.py b/...cated/datarobot_jdbc_batch_scoring_dag.py → ..._dags/datarobot_jdbc_batch_scoring_dag.py
diff --git a/.../deprecated/datarobot_jdbc_dataset_dag.py → ...xample_dags/datarobot_jdbc_dataset_dag.py b/.../deprecated/datarobot_jdbc_dataset_dag.py → ...xample_dags/datarobot_jdbc_dataset_dag.py
diff --git a/...ted/datarobot_jdbc_dynamic_dataset_dag.py → ...ags/datarobot_jdbc_dynamic_dataset_dag.py b/...ted/datarobot_jdbc_dynamic_dataset_dag.py → ...ags/datarobot_jdbc_dynamic_dataset_dag.py
diff --git a/...dags/deprecated/datarobot_pipeline_dag.py → ...al/example_dags/datarobot_pipeline_dag.py b/...dags/deprecated/datarobot_pipeline_dag.py → ...al/example_dags/datarobot_pipeline_dag.py
diff --git a/...obot_replace_custom_model_pipeline_dag.py → ...obot_replace_custom_model_pipeline_dag.py b/...obot_replace_custom_model_pipeline_dag.py → ...obot_replace_custom_model_pipeline_dag.py
diff --git a/...le_dags/deprecated/datarobot_score_dag.py → ...ental/example_dags/datarobot_score_dag.py b/...le_dags/deprecated/datarobot_score_dag.py → ...ental/example_dags/datarobot_score_dag.py
diff --git a/...ated/datarobot_timeseries_pipeline_dag.py → ...dags/datarobot_timeseries_pipeline_dag.py b/...ated/datarobot_timeseries_pipeline_dag.py → ...dags/datarobot_timeseries_pipeline_dag.py
diff --git a/...d/datarobot_upload_actuals_catalog_dag.py → ...s/datarobot_upload_actuals_catalog_dag.py b/...d/datarobot_upload_actuals_catalog_dag.py → ...s/datarobot_upload_actuals_catalog_dag.py
diff --git a/...ted/deployment_activate_deactivate_dag.py → ...ags/deployment_activate_deactivate_dag.py b/...ted/deployment_activate_deactivate_dag.py → ...ags/deployment_activate_deactivate_dag.py
diff --git a/...eprecated/deployment_replace_model_dag.py → ...mple_dags/deployment_replace_model_dag.py b/...eprecated/deployment_replace_model_dag.py → ...mple_dags/deployment_replace_model_dag.py
diff --git a/...eprecated/deployment_service_stats_dag.py → ...mple_dags/deployment_service_stats_dag.py b/...eprecated/deployment_service_stats_dag.py → ...mple_dags/deployment_service_stats_dag.py
diff --git a/...cated/deployment_stat_and_accuracy_dag.py → ..._dags/deployment_stat_and_accuracy_dag.py b/...cated/deployment_stat_and_accuracy_dag.py → ..._dags/deployment_stat_and_accuracy_dag.py
diff --git a/...loyment_update_monitoring_settings_dag.py → ...loyment_update_monitoring_settings_dag.py b/...loyment_update_monitoring_settings_dag.py → ...loyment_update_monitoring_settings_dag.py
diff --git a/...t_update_segment_analysis_settings_dag.py → ...t_update_segment_analysis_settings_dag.py b/...t_update_segment_analysis_settings_dag.py → ...t_update_segment_analysis_settings_dag.py
diff --git a/...nload_scoring_code_from_deployment_dag.py → ...nload_scoring_code_from_deployment_dag.py b/...nload_scoring_code_from_deployment_dag.py → ...nload_scoring_code_from_deployment_dag.py
diff --git a/.../deprecated/model_compute_insights_dag.py → ...xample_dags/model_compute_insights_dag.py b/.../deprecated/model_compute_insights_dag.py → ...xample_dags/model_compute_insights_dag.py
diff --git a/...el_compute_prediction_explanations_dag.py → ...el_compute_prediction_explanations_dag.py b/...el_compute_prediction_explanations_dag.py → ...el_compute_prediction_explanations_dag.py
diff --git a/...precated/model_compute_predictions_dag.py → ...ple_dags/model_compute_predictions_dag.py b/...precated/model_compute_predictions_dag.py → ...ple_dags/model_compute_predictions_dag.py
diff --git a/...dags/deprecated/model_compute_shap_dag.py → ...al/example_dags/model_compute_shap_dag.py b/...dags/deprecated/model_compute_shap_dag.py → ...al/example_dags/model_compute_shap_dag.py
diff --git a/...mple_dags/deprecated/model_retrain_dag.py → ...imental/example_dags/model_retrain_dag.py b/...mple_dags/deprecated/model_retrain_dag.py → ...imental/example_dags/model_retrain_dag.py
diff --git a/...xample_dags/deprecated/model_train_dag.py → ...erimental/example_dags/model_train_dag.py b/...xample_dags/deprecated/model_train_dag.py → ...erimental/example_dags/model_train_dag.py
diff --git a/requirements.txt b/requirements.txt
@@ -17,4 +17,4 @@ myst-parser==3.0.1
 black==24.10.0
 pyyaml>=6.0.2
 types-PyYAML>=6.0.12
-freezegun
+freezegun>=1.5.1
diff --git a/tests/unit/_experimental/__init__.py b/tests/unit/_experimental/__init__.py
diff --git a/tests/unit/dags/__init__.py → tests/unit/_experimental/dags/__init__.py b/tests/unit/dags/__init__.py → tests/unit/_experimental/dags/__init__.py
diff --git a/tests/unit/dags/conftest.py → tests/unit/_experimental/dags/conftest.py b/tests/unit/dags/conftest.py → tests/unit/_experimental/dags/conftest.py
diff --git a/tests/unit/dags/pipline/__init__.py → ...it/_experimental/dags/pipline/__init__.py b/tests/unit/dags/pipline/__init__.py → ...it/_experimental/dags/pipline/__init__.py
diff --git a/...gs/pipline/test_datarobot_pipeline_dag.py → ...gs/pipline/test_datarobot_pipeline_dag.py b/...gs/pipline/test_datarobot_pipeline_dag.py → ...gs/pipline/test_datarobot_pipeline_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_pipeline_dag import datarobot_pipeline
+from datarobot_provider._experimental.example_dags.datarobot_pipeline_dag import datarobot_pipeline
 
 
 def test_dag_loaded(dagbag):

diff --git a/tests/unit/dags/predictions/__init__.py → ...experimental/dags/predictions/__init__.py b/tests/unit/dags/predictions/__init__.py → ...experimental/dags/predictions/__init__.py
diff --git a/...arobot_azure_storage_batch_scoring_dag.py → ...arobot_azure_storage_batch_scoring_dag.py b/...arobot_azure_storage_batch_scoring_dag.py → ...arobot_azure_storage_batch_scoring_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_azure_storage_batch_scoring_dag import (
+from datarobot_provider._experimental.example_dags.datarobot_azure_storage_batch_scoring_dag import (
     datarobot_azure_storage_batch_scoring,
 )
 

diff --git a/...atarobot_gcp_storage_batch_scoring_dag.py → ...atarobot_gcp_storage_batch_scoring_dag.py b/...atarobot_gcp_storage_batch_scoring_dag.py → ...atarobot_gcp_storage_batch_scoring_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_gcp_storage_batch_scoring_dag import (
+from datarobot_provider._experimental.example_dags.datarobot_gcp_storage_batch_scoring_dag import (
     datarobot_gcp_batch_scoring,
 )
 

diff --git a/...ns/test_datarobot_s3_batch_scoring_dag.py → ...ns/test_datarobot_s3_batch_scoring_dag.py b/...ns/test_datarobot_s3_batch_scoring_dag.py → ...ns/test_datarobot_s3_batch_scoring_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_aws_s3_batch_scoring_dag import (
+from datarobot_provider._experimental.example_dags.datarobot_aws_s3_batch_scoring_dag import (
     datarobot_s3_batch_scoring,
 )
 

diff --git a/...s/predictions/test_datarobot_score_dag.py → ...s/predictions/test_datarobot_score_dag.py b/...s/predictions/test_datarobot_score_dag.py → ...s/predictions/test_datarobot_score_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_score_dag import datarobot_score
+from datarobot_provider._experimental.example_dags.datarobot_score_dag import datarobot_score
 
 
 def test_dag_loaded(dagbag):

diff --git a/tests/unit/dags/project/__init__.py → ...it/_experimental/dags/project/__init__.py b/tests/unit/dags/project/__init__.py → ...it/_experimental/dags/project/__init__.py
diff --git a/...est_create_project_from_ai_catalog_dag.py → ...est_create_project_from_ai_catalog_dag.py b/...est_create_project_from_ai_catalog_dag.py → ...est_create_project_from_ai_catalog_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_create_project_from_ai_catalog_dag import (
+from datarobot_provider._experimental.example_dags.datarobot_create_project_from_ai_catalog_dag import (
     create_project_from_aicatalog,
 )
 

diff --git a/...reate_project_from_dataset_version_dag.py → ...reate_project_from_dataset_version_dag.py b/...reate_project_from_dataset_version_dag.py → ...reate_project_from_dataset_version_dag.py
@@ -7,7 +7,7 @@
 # Released under the terms of DataRobot Tool and Utility Agreement.
 import pytest
 
-from datarobot_provider.example_dags.deprecated.datarobot_create_project_from_dataset_version_dag import (
+from datarobot_provider._experimental.example_dags.datarobot_create_project_from_dataset_version_dag import (
     create_project_from_dataset_version,
 )