Add for autocompletion in ipython

Eden-Kramer-Lab · Mar 23, 2018 · 70c832b · 70c832b
1 parent b8bf7cf
commit 70c832b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/replay_classification/decoders.py b/replay_classification/decoders.py
@@ -83,6 +83,9 @@ def __init__(self, position, lagged_position,
         self.place_std_deviation = place_std_deviation
         self.confidence_threshold = confidence_threshold
 
+    def __dir__(self):
+        return self.keys()
+
     def fit(self):
         '''Fits the decoder model for each trajectory_direction.
 
@@ -291,6 +294,9 @@ def __init__(self, position, lagged_position, spikes, trajectory_direction,
         self.confidence_threshold = confidence_threshold
         self.knot_spacing = knot_spacing
 
+    def __dir__(self):
+        return self.keys()
+
     def fit(self):
         '''Fits the decoder model by state