Labelled show_selectors as optional

TransformerLensOrg · Sep 20, 2023 · c6dcf2e · c6dcf2e
1 parent 113d10d
commit c6dcf2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/circuitsvis/activations.py b/python/circuitsvis/activations.py
@@ -15,7 +15,7 @@ def text_neuron_activations(
     second_dimension_labels: Optional[List[str]] = None,
     first_dimension_default: Optional[int] = 0,
     second_dimension_default: Optional[int] = 0,
-    show_selectors: bool = True,
+    show_selectors: Optional[bool] = True,
 ) -> RenderedHTML:
     """Show activations (colored by intensity) for each token in a text or set
     of texts.