diff --git a/pyproject.toml b/pyproject.toml index 3bd45c2..0e4f15a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -21,4 +21,4 @@ readme = "README.md" [tool.setuptools] package-dir = {"" = "src"} -packages = ["rbm", "image", "stack"] +packages = ["rbm", "image", "stack", "label"] diff --git a/src/label/__init__.py b/src/label/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/rbm/label.py b/src/label/label.py similarity index 100% rename from src/rbm/label.py rename to src/label/label.py diff --git a/src/tests/test_label.py b/src/tests/test_label.py index 2124edb..dddf0da 100644 --- a/src/tests/test_label.py +++ b/src/tests/test_label.py @@ -1,7 +1,7 @@ import matplotlib.pyplot as plt from rbm.matrix import np -from rbm.label import Label +from label.label import Label if __name__ == "__main__": voc_size = 100 diff --git a/src/tests/test_learn_encoded_labels.py b/src/tests/test_learn_encoded_labels.py index 8aefde8..5e6f598 100644 --- a/src/tests/test_learn_encoded_labels.py +++ b/src/tests/test_learn_encoded_labels.py @@ -2,7 +2,7 @@ import matplotlib.pyplot as plt from rbm.model import Model -from rbm.label import Label +from label.label import Label from rbm.entity import Entity, EntityParams, TrainingParams from rbm.matrix import Mat, np diff --git a/src/tests/test_learn_norbs_labels.py b/src/tests/test_learn_norbs_labels.py index d27d06c..ac0cc33 100644 --- a/src/tests/test_learn_norbs_labels.py +++ b/src/tests/test_learn_norbs_labels.py @@ -4,7 +4,7 @@ from rbm.model import Model from rbm.entity import Entity, EntityParams, TrainingParams from rbm.matrix import Mat, np, read_armadillo from rbm.train import train, Status -from rbm.label import Label +from label.label import Label class TestModel(Model): def __init__(self, name: str, work_dir: str = '.'):