diff --git a/research/research.py b/research/research.py index 0f2fa6a..14961c5 100644 --- a/research/research.py +++ b/research/research.py @@ -16,13 +16,7 @@ # %% from typing import Optional -# %% # %% tags=["parameters"] -data_path: Optional[str] = None -# Полный путь к файлу (CSV) с исходным датасетом. Если не установлен, ищется файл в `data/`. -data_relpath: str = 'cars.csv' -# Путь к файлу (CSV) с исходным датасетом относительно директории данных `data`. Игнорируется, если установлен data_path. - data_aug_pickle_path: Optional[str] = None # Полный путь к файлу (pickle) для сохранения очищенного датасета. Если не установлен, используется `data/`. data_aug_pickle_relpath: str = 'cars.aug.pickle' @@ -61,8 +55,8 @@ if mlflow_registry_uri is not None: # %% DATA_PATH = ( - pathlib.Path(os.path.dirname(data_path)) - if data_path is not None + pathlib.Path(os.path.dirname(data_aug_pickle_path)) + if data_aug_pickle_path is not None else (BASE_PATH / 'data') )