From 3162319f935b131f64e16f0ab1e8e09ca41c9ee4 Mon Sep 17 00:00:00 2001 From: Finn Sherry Date: Wed, 15 Oct 2025 17:19:21 +0200 Subject: [PATCH] move diffusion-shock models to experiments directory to avoid adding dependencies to lietorch. --- {lietorch/models => experiments}/ds_pde.py | 0 experiments/run_ds_pde.py | 11 ++++++----- lietorch/models/__init__.py | 1 - 3 files changed, 6 insertions(+), 6 deletions(-) rename {lietorch/models => experiments}/ds_pde.py (100%) diff --git a/lietorch/models/ds_pde.py b/experiments/ds_pde.py similarity index 100% rename from lietorch/models/ds_pde.py rename to experiments/ds_pde.py diff --git a/experiments/run_ds_pde.py b/experiments/run_ds_pde.py index 6ba5781..444851d 100644 --- a/experiments/run_ds_pde.py +++ b/experiments/run_ds_pde.py @@ -1,4 +1,4 @@ -from lietorch.models import ds_pde +from . import ds_pde from dataclasses import asdict @@ -105,11 +105,12 @@ if __name__ == "__main__": parser.add_argument( "--runname", type=str, default="", help="Name of the training run." ) + parser.add_argument("--tag", type=str, default=None, help="wandb tag.") parser.add_argument( - "--tag", type=str, default=None, help="wandb tag." - ) - parser.add_argument( - "--fixed_lift", type=eval, default=True, help="Lift with cakewavelets or trained wavelets." + "--fixed_lift", + type=eval, + default=True, + help="Lift with cakewavelets or trained wavelets.", ) args = parser.parse_args() diff --git a/lietorch/models/__init__.py b/lietorch/models/__init__.py index e6f50fa..ae25a5c 100644 --- a/lietorch/models/__init__.py +++ b/lietorch/models/__init__.py @@ -6,4 +6,3 @@ import lietorch.models.dca1 import lietorch.models.drive import lietorch.models.rotnist import lietorch.models.fixed_lift -import lietorch.models.ds_pde -- GitLab