Minor cleanup

Denys88 · Oct 27, 2024 · a28f62a · a28f62a
1 parent 2c05bcc
commit a28f62a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 176 deletions.
diff --git a/rl_games/networks/__init__.py b/rl_games/networks/__init__.py
@@ -1,7 +1,4 @@
 from rl_games.networks.tcnn_mlp import TcnnNetBuilder
-from rl_games.networks.moe import MoENetBuilder
-
 from rl_games.algos_torch import model_builder
 
-model_builder.register_network('tcnnnet', TcnnNetBuilder)
-model_builder.register_network('moe', MoENetBuilder)
+model_builder.register_network('tcnnnet', TcnnNetBuilder)
diff --git a/rl_games/networks/moe.py b/rl_games/networks/moe.py