Annotate the memorized coefficient maps

dtree_learner.py

@@ -3,7 +3,7 @@ import csv
 import itertools
 import json
 import os
-from typing import Any, List, Optional, Tuple
+from typing import Any, Dict, List, Optional, Tuple
 
 import numpy as np
 
@@ -19,7 +19,10 @@ class DTreeLearner(LearnerBase):
         self._perc_dim: int = perc_dim
 
         self._s2f_func = lambda x: x
-        self._var_coeff_map = {}
+
+        # Given a (derived) feature name,
+        # this map returns a mapping from base feature names to coefficients
+        self._var_coeff_map: Dict[str, Dict[str, int]] = {}
 
         # check directory name exists, if not create it.
         self.dir_name = "out"