lint

pytorch · Apr 18, 2024 · 7441f79 · 7441f79
1 parent 4451f63
commit 7441f79
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/sota-implementations/dreamer/dreamer.py b/sota-implementations/dreamer/dreamer.py
@@ -22,7 +22,7 @@
 # mixed precision training
 from torch.cuda.amp import GradScaler
 from torch.nn.utils import clip_grad_norm_
-from torch.profiler import profile, ProfilerActivity, record_function
+from torch.profiler import profile, ProfilerActivity
 from torchrl._utils import logger as torchrl_logger, timeit
 from torchrl.envs.utils import ExplorationType, set_exploration_type
 from torchrl.modules import RSSMRollout

diff --git a/sota-implementations/dreamer/dreamer_utils.py b/sota-implementations/dreamer/dreamer_utils.py
@@ -35,7 +35,6 @@
     DoubleToFloat,
     DreamerDecoder,
     DreamerEnv,
-    DTypeCastTransform,
     EnvCreator,
     ExcludeTransform,
     # ExcludeTransform,