From ebd2c276b80577c0234001c37a8f26b1d07dff37 Mon Sep 17 00:00:00 2001 From: LostOxygen Date: Mon, 10 Jun 2024 13:56:03 +0200 Subject: [PATCH] fix: Module not found and import error for experiments fixed --- experiments/__init__.py | 0 .../data/formal_language/formal_language_dataset.py | 2 +- experiments/{ => data/formal_language}/metrics.py | 0 experiments/main.py | 11 +++++------ 4 files changed, 6 insertions(+), 7 deletions(-) create mode 100644 experiments/__init__.py rename experiments/{ => data/formal_language}/metrics.py (100%) diff --git a/experiments/__init__.py b/experiments/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/experiments/data/formal_language/formal_language_dataset.py b/experiments/data/formal_language/formal_language_dataset.py index c5d6520..15c75e4 100644 --- a/experiments/data/formal_language/formal_language_dataset.py +++ b/experiments/data/formal_language/formal_language_dataset.py @@ -12,7 +12,7 @@ from .generate import ALL_ARGS from .online_generate import OnlineTaskGenerateMaskedSeparate from ..utils import DataGen -from ...metrics import SequenceAccuracy +from .metrics import SequenceAccuracy def create_dataclass(cls_name, params_list, additional=[]): diff --git a/experiments/metrics.py b/experiments/data/formal_language/metrics.py similarity index 100% rename from experiments/metrics.py rename to experiments/data/formal_language/metrics.py diff --git a/experiments/main.py b/experiments/main.py index 14d757f..8379727 100644 --- a/experiments/main.py +++ b/experiments/main.py @@ -6,18 +6,17 @@ import torch import torch.optim as optim from dacite import from_dict -from experiments.data.formal_language.formal_language_dataset import ( - FormLangDatasetGenerator, -) -from experiments.data.utils import DataGen -from experiments.lr_scheduler import LinearWarmupCosineAnnealing from omegaconf import DictConfig, OmegaConf from torch import nn from torch.utils.data import DataLoader from tqdm import tqdm - from xlstm.xlstm_lm_model import xLSTMLMModel, xLSTMLMModelConfig +from data.formal_language.formal_language_dataset import \ + FormLangDatasetGenerator +from data.utils import DataGen +from lr_scheduler import LinearWarmupCosineAnnealing + dataset_registry: dict[str, Type[DataGen]] = { "form_language": FormLangDatasetGenerator }