From 196418fc734092dac4ab31547892ac0e5be579ec Mon Sep 17 00:00:00 2001
From: Isaac Virshup <ivirshup@gmail.com>
Date: Thu, 11 Jul 2024 17:57:10 -0700
Subject: [PATCH] Fix doc versions (#1238)

---
 docs/conf.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/docs/conf.py b/docs/conf.py
index f4d63e0ed..a1d2c127f 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -10,13 +10,13 @@
 copyright = '2022, Chan Zuckerberg Initiative'
 author = 'Chan Zuckerberg Initiative'
 
+from packaging.version import Version
 import git
+
 repo = git.Repo(search_parent_directories=True)
-tags = [t for t in repo.tags if t.tag is not None]
-tags = sorted(tags, key=lambda t: t.tag.tagged_date)
-latest_tag = tags[-1]
+all_versions = sorted([Version(t.name) for t in repo.tags])
 
-version = str(latest_tag)
+version = str(all_versions[-1])
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
@@ -28,7 +28,7 @@
     'sphinx.ext.napoleon', 
     'sphinx.ext.autosummary', 
     'myst_parser'
-    ]
+]
 
 autosummary_generate = True