Skip to content
This repository was archived by the owner on Jun 17, 2024. It is now read-only.

Commit 833a4d6

Browse files
fix: Fix jinja to handle file paths as raw strings (#6)
Signed-off-by: arima-tsukasa <[email protected]>
1 parent bd45adb commit 833a4d6

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

sapientml_loaddata/templates/loaddata.py.jinja

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,24 +2,24 @@
22
import pandas as pd
33

44
{% if dataset.training_data_path.endswith(".pkl") %}
5-
train_dataset = pd.read_pickle("{{ dataset.training_data_path }}")
5+
train_dataset = pd.read_pickle(r"{{ dataset.training_data_path }}")
66
{% else %}
7-
train_dataset = pd.read_csv("{{ dataset.training_data_path }}", encoding="{{ dataset.csv_encoding }}", delimiter="{{ dataset.csv_delimiter }}")
7+
train_dataset = pd.read_csv(r"{{ dataset.training_data_path }}", encoding="{{ dataset.csv_encoding }}", delimiter="{{ dataset.csv_delimiter }}")
88
{% endif %}
99

1010
{% if dataset.validation_data_path %}
1111
{% if dataset.validation_data_path.endswith(".pkl") %}
12-
validation_dataset = pd.read_pickle("{{ dataset.validation_data_path }}")
12+
validation_dataset = pd.read_pickle(r"{{ dataset.validation_data_path }}")
1313
{% else %}
14-
validation_dataset = pd.read_csv("{{ dataset.validation_data_path }}", encoding="{{ dataset.csv_encoding }}", delimiter="{{ dataset.csv_delimiter }}")
14+
validation_dataset = pd.read_csv(r"{{ dataset.validation_data_path }}", encoding="{{ dataset.csv_encoding }}", delimiter="{{ dataset.csv_delimiter }}")
1515
{% endif %}
1616
{% endif %}{# if dataset.validation_data_path #}
1717

1818
{% if not validation and dataset.test_data_path %}
1919
{% if dataset.test_data_path.endswith(".pkl") %}
20-
test_dataset = pd.read_pickle("{{ dataset.test_data_path }}")
20+
test_dataset = pd.read_pickle(r"{{ dataset.test_data_path }}")
2121
{% else %}
22-
test_dataset = pd.read_csv("{{ dataset.test_data_path }}", encoding="{{ dataset.csv_encoding }}", delimiter="{{ dataset.csv_delimiter }}")
22+
test_dataset = pd.read_csv(r"{{ dataset.test_data_path }}", encoding="{{ dataset.csv_encoding }}", delimiter="{{ dataset.csv_delimiter }}")
2323
{% endif %}
2424
{% endif %}{# if not validation and dataset.test_data_path #}
2525

0 commit comments

Comments
 (0)