Fixes inheritance issue.

canvas-medical · Nov 26, 2024 · 337c1a0 · 337c1a0
1 parent bc02e8d
commit 337c1a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/canvas_sdk/v1/data/base.py b/canvas_sdk/v1/data/base.py
@@ -158,13 +158,13 @@ def within(self, timeframe: "Timeframe") -> models.QuerySet:
         )
 
 
-class ValueSetLookupQuerySet(BaseQuerySet, ValueSetLookupQuerySetMixin):
+class ValueSetLookupQuerySet(CommittableQuerySet, ValueSetLookupQuerySetMixin):
     """A class that includes methods for looking up value sets."""
 
     pass
 
 
-class ValueSetLookupByNameQuerySet(BaseQuerySet, ValueSetLookupByNameQuerySetMixin):
+class ValueSetLookupByNameQuerySet(CommittableQuerySet, ValueSetLookupByNameQuerySetMixin):
     """A class that includes methods for looking up value sets by name."""
 
     pass

diff --git a/canvas_sdk/v1/data/condition.py b/canvas_sdk/v1/data/condition.py
@@ -1,11 +1,7 @@
 from django.db import models
 from django.db.models import TextChoices
 
-from canvas_sdk.v1.data.base import (
-    CommittableModelManager,
-    CommittableQuerySet,
-    ValueSetLookupQuerySet,
-)
+from canvas_sdk.v1.data.base import ValueSetLookupQuerySet
 from canvas_sdk.v1.data.patient import Patient
 from canvas_sdk.v1.data.user import CanvasUser