diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index bf8ec1e..bb8647c 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -11,7 +11,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.6", "3.7", "3.8", "3.9"]
+        python-version: ["3.7", "3.8", "3.9"]
 
     steps:
     - uses: actions/checkout@v2
diff --git a/torchinfo/__init__.py b/torchinfo/__init__.py
index de2395d..b078c3c 100644
--- a/torchinfo/__init__.py
+++ b/torchinfo/__init__.py
@@ -4,4 +4,4 @@
 from .torchinfo import summary
 
 __all__ = ("ModelStatistics", "summary", "ALL_COLUMN_SETTINGS", "ALL_ROW_SETTINGS")
-__version__ = "1.5.4"
+__version__ = "1.6.0"