Update

[ghstack-poisoned]
pytorch · Dec 16, 2024 · 545bcc0 · 545bcc0
1 parent 9cbe008
commit 545bcc0
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/sota-implementations/td3/utils.py b/sota-implementations/td3/utils.py
@@ -29,12 +29,7 @@
 )
 from torchrl.envs.libs.gym import GymEnv, set_gym_backend
 from torchrl.envs.utils import ExplorationType, set_exploration_type
-from torchrl.modules import (
-    AdditiveGaussianModule,
-    MLP,
-    TanhModule,
-    ValueOperator,
-)
+from torchrl.modules import AdditiveGaussianModule, MLP, TanhModule, ValueOperator
 
 from torchrl.objectives import SoftUpdate
 from torchrl.objectives.td3 import TD3Loss