fix typo

eltociear · Nov 5, 2024 · 6ae47c5 · 6ae47c5
1 parent a6ade22
commit 6ae47c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/common/registry/test_registry.py b/tests/common/registry/test_registry.py
@@ -2,7 +2,7 @@
 
 from typing import List, Dict, Union
 from pyhocon import ConfigTree, ConfigFactory
-from kag_ant.common.registry import Registrable, Lazy, Functor
+from kag.common.registry import Registrable, Lazy, Functor
 import numpy as np
 
 
@@ -13,7 +13,7 @@ def __init__(self, name: str = "mock_model"):
 
 @MockModel.register("Simple")
 class Simple(MockModel):
-    def __init__(self, name, age=999):
+    def __init__(self, name, age=None):
         pass