fixed context manager bug

Datagration · May 3, 2024 · e7ee1f1 · e7ee1f1
1 parent 64acf5f
commit e7ee1f1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/petrovisor/__init__.py b/src/petrovisor/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.1.6"
+__version__ = "0.1.7"
 
 # api
 from petrovisor.petrovisor import PetroVisor

diff --git a/src/petrovisor/models/contexts_manager.py b/src/petrovisor/models/contexts_manager.py
@@ -103,11 +103,14 @@ def __init__(
 
         if len(all_contexts) == 0:
             context = Context(name=default_name or "")
+            context.entity_set = entity_set
+            context.scope = scope
+            context.hierarchy = hierarchy
         elif len(all_contexts) == 1 or primary_context.casefold() == "first":
             context = all_contexts[0]
         else:
             context = api.merge_contexts(all_contexts)
         self.name = context.name
-        self.entity_set = context.entity_set
-        self.scope = context.scope
-        self.hierarchy = context.hierarchy
+        self.entity_set = context.entity_set or entity_set
+        self.scope = context.scope or scope
+        self.hierarchy = context.hierarchy or hierarchy