diff options
| -rw-r--r-- | tests/test_core.py | 4 | ||||
| -rw-r--r-- | warlock/core.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_core.py b/tests/test_core.py index daa458c..923d94f 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -260,8 +260,8 @@ class TestCore(unittest.TestCase): country_schema = json.load(country_schema_file) person_schema = json.load(person_schema_file) - Country = warlock.model_factory(country_schema, resolver) - Person = warlock.model_factory(person_schema, resolver) + Country = warlock.model_factory(country_schema, resolver=resolver) + Person = warlock.model_factory(person_schema, resolver=resolver) england = Country( name="England", diff --git a/warlock/core.py b/warlock/core.py index 621b0a8..a6dd3a4 100644 --- a/warlock/core.py +++ b/warlock/core.py @@ -19,7 +19,7 @@ import copy from . import model -def model_factory(schema, resolver=None, base_class=model.Model, name=None): +def model_factory(schema, base_class=model.Model, name=None, resolver=None): """Generate a model class based on the provided JSON Schema :param schema: dict representing valid JSON schema |