Skip to content

Commit 144b6ef

Browse files
authored
[Ez] Fix broken actor import __init__ (#426)
1 parent 25d8a7a commit 144b6ef

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

src/forge/actors/__init__.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,23 +5,19 @@
55
# LICENSE file in the root directory of this source tree.
66

77
__all__ = [
8-
"Generator" "PolicyRouter",
8+
"Generator",
99
"RLTrainer",
1010
"ReplayBuffer",
11-
"TitanRefModel",
11+
"ReferenceModel",
1212
"SandboxedPythonCoder",
1313
]
1414

1515

1616
def __getattr__(name):
1717
if name == "Generator":
18-
from .policy import Generator
18+
from .generator import Generator
1919

2020
return Generator
21-
elif name == "PolicyRouter":
22-
from .policy import PolicyRouter
23-
24-
return PolicyRouter
2521
elif name == "RLTrainer":
2622
from .trainer import RLTrainer
2723

0 commit comments

Comments
 (0)