Skip to content

Commit

Permalink
fix test for anomaly detection module
Browse files Browse the repository at this point in the history
  • Loading branch information
KGallyamov committed May 19, 2024
1 parent 80ff952 commit f0c9484
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from tests.fixtures.config import models as fixt_models
from tests.fixtures.config import optimizers as fixt_optimizers
from tests.fixtures.config import trainers as fixt_trainers
from tests.fixtures.config import schedulers as fixt_schedulers


def test_anomaly_detection():
Expand All @@ -27,9 +28,10 @@ def test_anomaly_detection():
model = get_model(cfg.models, cfg.trainer)
losses = get_losses(cfg, "anomaly-detection-timeseries", Frameworks.torch)
optimizer_cfg = DictConfig(fixt_optimizers.adam_optim_w_target)
scheduler_cfg = DictConfig(fixt_schedulers.linear_w_target)

module = AnomalyDetectionTimeSeriesLightningModule(
model=model, losses=losses, optimizer_cfg=optimizer_cfg
model=model, losses=losses, optimizer_cfg=optimizer_cfg, scheduler_cfg=scheduler_cfg
)

assert module is not None
Expand Down

0 comments on commit f0c9484

Please sign in to comment.