File tree Expand file tree Collapse file tree 2 files changed +11
-0
lines changed
Expand file tree Collapse file tree 2 files changed +11
-0
lines changed Original file line number Diff line number Diff line change @@ -60,3 +60,8 @@ cat_features:
6060
6161target :
6262 - fare_amount
63+
64+ train_file_name : " green_taxi_train_set.parquet"
65+ test_file_name : " green_taxi_test_set.parquet"
66+
67+ experiment_name : " taxi_fare_prediction"
Original file line number Diff line number Diff line change @@ -23,6 +23,9 @@ def valid_yaml_config():
2323 cat_features: []
2424 target:
2525 - fare_amount
26+ train_file_name: "train.csv"
27+ test_file_name: "test.csv"
28+ experiment_name: "my-experiment"
2629 """
2730
2831
@@ -60,6 +63,9 @@ def test_project_config_from_valid_yaml(valid_yaml_config):
6063 assert config .num_features == ["col1" , "col2" ]
6164 assert config .cat_features == []
6265 assert config .target == ["fare_amount" ]
66+ assert config .train_file_name == "train.csv"
67+ assert config .test_file_name == "test.csv"
68+ assert config .experiment_name == "my-experiment"
6369
6470
6571def test_project_config_missing_field (invalid_yaml_config_missing_field ):
You can’t perform that action at this time.
0 commit comments