Merge pull request #66 from jambit/dcs

aai-institute · May 17, 2023 · 034cf8a · 034cf8a
2 parents 0eccda2 + 641014d
commit 034cf8a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/sensai/util/aggregation.py b/src/sensai/util/aggregation.py
@@ -53,9 +53,13 @@ class DistributionCounter(ToStringMixin):
  Supports the counting of the frequencies with which (mutually exclusive) events occur
  """
  def __init__(self):
- self.counts = collections.defaultdict(lambda: 0)
+ self.counts = collections.defaultdict(self._zero)
  self.totalCount = 0
 
+ @staticmethod
+ def _zero():
+ return 0
+
  def count(self, event: Hashable) -> None:
  """
  Increments the count of the given event

diff --git a/src/sensai/util/logging.py b/src/sensai/util/logging.py
@@ -8,7 +8,6 @@
 
 import pandas as pd
 
-from sensai.util.deprecation import deprecated
 
 log = getLogger(__name__)