From 2a258828a51eb655394af855daa0c7193a92fd12 Mon Sep 17 00:00:00 2001
From: JohnMark Taylor <johnmarktaylor@g.harvard.edu>
Date: Thu, 22 Aug 2024 16:50:46 -0700
Subject: [PATCH] Fixed a bug where TorchLens wasn't accounting for the dim
 argument in torch.max and torch.min (they return a tuple subclass with the
 values and the indices).

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 66438d0..07ee51f 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
 
 setup(
     name="torchlens",
-    version="0.1.17",
+    version="0.1.18",
     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 "