Allow environment variables in dagster-mlflow schema (#7997)

* refactor: allow environment variables in dagster-mlflow schema * fix: isort
dagster-io · Jun 2, 2022 · 67e4b01 · 67e4b01
1 parent 04f9c4e
commit 67e4b01
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python_modules/libraries/dagster-mlflow/dagster_mlflow/resources.py b/python_modules/libraries/dagster-mlflow/dagster_mlflow/resources.py
@@ -12,12 +12,12 @@
 import mlflow
 from mlflow.entities.run_status import RunStatus
 
-from dagster import Field, Noneable, Permissive, resource
+from dagster import Field, Noneable, Permissive, StringSource, resource
 
 CONFIG_SCHEMA = {
-    "experiment_name": Field(str, is_required=True, description="MlFlow experiment name."),
+    "experiment_name": Field(StringSource, is_required=True, description="MlFlow experiment name."),
     "mlflow_tracking_uri": Field(
-        Noneable(str),
+        Noneable(StringSource),
         default_value=None,
         is_required=False,
         description="MlFlow tracking server uri.",