fix version comparison (#38)

pmeier · Aug 23, 2021 · 030771d · 030771d
1 parent 476610e
commit 030771d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/light_the_torch/_pip/find.py b/light_the_torch/_pip/find.py
@@ -251,7 +251,7 @@ def _sort_key(
             cb.ComputationBackend.from_str(
                 candidate.version.local.replace("any", "cpu")
             ),
-            candidate.version.base_version,
+            candidate.version,
         )
 
     def get_applicable_candidates(