remove constants

src/delphi/__init__.py

@@ -1,5 +1,10 @@
+from importlib.resources import files
+from pathlib import Path
+from typing import cast
+
 from beartype.claw import beartype_this_package  # <-- hype comes
 
 beartype_this_package()  # <-- hype goes
 
-__version__ = "0.1.1"
+__version__ = "0.2"
+TEST_CONFIGS_DIR = cast(Path, files("delphi.test_configs"))

src/delphi/utils.py

@@ -7,7 +7,6 @@ def hf_split_to_split_name(split: str) -> str:
     return split.split("[")[0]
 
 
-# TODO: test load_dataset functions
 def load_dataset_split_features(
     repo_id: str,
     split: str,

tests/train/config/test_config_utils.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from delphi.constants import TEST_CONFIGS_DIR
+from delphi import TEST_CONFIGS_DIR
 from delphi.train.config.utils import (
     _unoptionalize,
     build_config_from_files_and_overrides,

tests/train/test_train_step.py

@@ -7,7 +7,7 @@
 from jaxtyping import Float
 from transformers import PreTrainedModel
 
-from delphi.constants import TEST_CONFIGS_DIR
+from delphi import TEST_CONFIGS_DIR
 from delphi.eval.utils import get_all_and_next_logprobs
 from delphi.train.config import TrainingConfig
 from delphi.train.config.utils import build_config_from_files_and_overrides

tests/train/test_wandb_utils.py

@@ -7,7 +7,7 @@
 import transformers
 from dacite import from_dict
 
-from delphi.constants import TEST_CONFIGS_DIR
+from delphi import TEST_CONFIGS_DIR
 from delphi.train.config import TrainingConfig
 from delphi.train.config.utils import build_config_from_files_and_overrides
 from delphi.train.utils import ModelTrainingState, initialize_model_training_state