Fixed output labeling

johnmarktaylor91 · Sep 8, 2023 · f4517e8 · f4517e8
1 parent 317a70b
commit f4517e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -16,7 +16,7 @@
 
 setup(
     name="torchlens",
-    version="0.1.5",
+    version="0.1.6",
     description="A package for extracting activations from PyTorch models",
     long_description="A package for extracting activations from PyTorch models. Contains functionality for "
                      "extracting model activations, visualizing a model's computational graph, and "

diff --git a/torchlens/helper_funcs.py b/torchlens/helper_funcs.py
@@ -326,7 +326,7 @@ def extend_search_stack_from_item(
         item: The item
         next_stack: Stack to add to
     """
-    if issubclass(type(item), (list, tuple, set)):
+    if type(item) in [list, tuple, set]:
         if address == "":
             next_stack.extend(
                 [(x, f"{i}", address_full + [("ind", i)]) for i, x in enumerate(item)]