Fix init to accept kwargs

This commit is contained in:
Connor Shorten
2025-11-26 19:26:58 -05:00
parent 35a155ab16
commit 65f6880c6f

View File

@@ -9,20 +9,18 @@ class EchoConfig(PrecompiledConfig):
class EchoAgent(PrecompiledAgent):
config: EchoConfig
def __init__(self, config: EchoConfig):
super().__init__(config)
def __init__(self, config: EchoConfig, **kwargs): # <-- Add **kwargs
super().__init__(config, **kwargs) # <-- Pass them through
def forward(self, text: str) -> str:
return f"Echo: {text}"
# This block only runs when you execute the script directly,
# NOT when modaic imports the module
if __name__ == "__main__":
agent = EchoAgent(EchoConfig())
print(agent(text="hello"))
agent.push_to_hub(
"connor/CrossEncoderRanker",
with_code=True,
commit_message="My first commit"
commit_message="Fix init to accept kwargs"
)