fixes

src/dbally/exceptions.py

@@ -2,10 +2,3 @@ class DbAllyError(Exception):
     """
     Base class for all exceptions raised by db-ally.
     """
-
-
-class UnsupportedAggregationError(DbAllyError):
-    """
-    Error raised when AggregationFormatter is unable to construct a query
-    with given aggregation.
-    """

src/dbally/views/structured.py

@@ -1,6 +1,6 @@
 import abc
 from collections import defaultdict
-from typing import Any, Dict, List, Optional, TypeVar
+from typing import Dict, List, Optional
 
 from dbally.audit.event_tracker import EventTracker
 from dbally.collection.results import ViewExecutionResult
@@ -14,8 +14,6 @@
 from ..similarity import AbstractSimilarityIndex
 from .base import BaseView, IndexLocation
 
-DataT = TypeVar("DataT", bound=Any)
-
 
 class BaseStructuredView(BaseView):
     """

tests/unit/views/test_pandas_base.py

@@ -41,19 +41,19 @@ class MockDataFrameView(DataFrameBaseView):
 
     @view_filter()
     def filter_city(self, city: str) -> pd.Series:
-        return self.data["city"] == city
+        return self.df["city"] == city
 
     @view_filter()
     def filter_year(self, year: int) -> pd.Series:
-        return self.data["year"] == year
+        return self.df["year"] == year
 
     @view_filter()
     def filter_age(self, age: int) -> pd.Series:
-        return self.data["age"] == age
+        return self.df["age"] == age
 
     @view_filter()
     def filter_name(self, name: str) -> pd.Series:
-        return self.data["name"] == name
+        return self.df["name"] == name
 
     @view_aggregation()
     def mean_age_by_city(self) -> AggregationGroup: