Skip to content

Commit 34ab501

Browse files
committed
update project config with new fields
1 parent d2b9abf commit 34ab501

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

project-config.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,3 +60,8 @@ cat_features:
6060

6161
target:
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"

tests/test_project_config.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff 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

6571
def test_project_config_missing_field(invalid_yaml_config_missing_field):

0 commit comments

Comments
 (0)