use _ytBaseModel

data-exp-lab · Feb 12, 2024 · f5107b3 · f5107b3
1 parent 5fe2425
commit f5107b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/yt_napari/_data_model.py b/src/yt_napari/_data_model.py
@@ -66,7 +66,7 @@ class Region(_ytBaseModel):
     rescale: bool = Field(False, description="rescale the final image between 0,1")
 
 
-class CoveringGrid(BaseModel):
+class CoveringGrid(_ytBaseModel):
     fields: List[ytField] = Field(
         None, description="list of fields to load for this selection"
     )
@@ -86,7 +86,7 @@ class CoveringGrid(BaseModel):
     rescale: bool = Field(False, description="rescale the final image between 0,1")
 
 
-class Slice(BaseModel):
+class Slice(_ytBaseModel):
     fields: List[ytField] = Field(
         None, description="list of fields to load for this selection"
     )