Create an experimental directory for which we don't apply the formatt…

…ing rules.
dynamicslab · Oct 27, 2024 · c690987 · c690987
1 parent 94a22cf
commit c690987
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/hydrogym/experimental/__init__.py b/hydrogym/experimental/__init__.py
diff --git a/hydrogym/core_1DEnvs.py → hydrogym/experimental/core_1DEnvs.py b/hydrogym/core_1DEnvs.py → hydrogym/experimental/core_1DEnvs.py
@@ -136,7 +136,6 @@ def evaluate_objective(self) -> ArrayLike:
     pass
 
 if __name__ == '__main__':
-    # torch.set_default_tensor_type('torch.cuda.FloatTensor')
     from tqdm import tqdm
     import torch
 
@@ -154,7 +153,6 @@ def evaluate_objective(self) -> ArrayLike:
             "max_steps": 100,
         }
 
-    # ks_init = torch.load('/net/beegfs-hpc/work/lagemannk/container/hydrogym_dev2/home/firedrake/hydrogym_rllib_multiEnv/hydrogym/hydrogym/torch_env/ks_init.tensor')
     env = OneDimEnv(config)
 
     action_space = env.action_space 
@@ -186,9 +184,5 @@ def evaluate_objective(self) -> ArrayLike:
     plt.savefig('/net/beegfs-hpc/work/lagemannk/container/workspace_christian_ext/logs/result.png', bbox_inches='tight')
     plt.close()
 
-    env.reset()
-
-
-
-
+    env.reset()