update config files

victorkaillo · Feb 22, 2022 · ed8293b · ed8293b
1 parent 0d714c5
commit ed8293b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/check_data/conftest.py b/check_data/conftest.py
@@ -42,4 +42,4 @@ def ks_alpha(request):
 
 # mlflow run . -P reference_artifact="mlops-week_09_Example_01_segregate/train_data.csv:latest" \
 #              -P sample_artifact="mlops-week_09_Example_01_segregate/test_data.csv:latest" \
-#              -P ks_alpha=0.05
+#              -P ks_a
diff --git a/config.yaml b/config.yaml
@@ -15,7 +15,7 @@ main:
 data:
   train_data: "mlops-creditcard_fraud_predictive/train_data.csv:latest"
   file_url: "https://www.kaggle.com/mlg-ulb/creditcardfraud.csv"
-  reference_dataset: "mlops-creditcard_fraud_predictive/ccfraud_preprocessed.csv:latest"
+  reference_dataset: "mlops-creditcard_fraud_predictive/raw_data.csv:latest"
   # Threshold for Kolomorov-Smirnov test
   ks_alpha: 0.05
   test_size: 0.3

diff --git a/main.py b/main.py
@@ -53,7 +53,7 @@ def process_args(config: DictConfig):
             "main",
             parameters={
                 "reference_artifact": config["data"]["reference_dataset"],
-                "sample_artifact": "preprocessed_data.csv:latest",
+                "sample_artifact": "raw_data.csv:latest",
                 "ks_alpha": config["data"]["ks_alpha"]
             }
         )
@@ -64,7 +64,7 @@ def process_args(config: DictConfig):
             os.path.join(root_path, "segregate"),
             "main",
             parameters={
-                "input_artifact": "preprocessed_data.csv:latest",
+                "input_artifact": "raw_data.csv:latest",
                 "artifact_root": "data",
                 "artifact_type": "segregated_data",
                 "test_size": config["data"]["test_size"],