fix ROCm deselection (#56)

Author: pmeier

light_the_torch/_patch.py

@@ -244,8 +244,11 @@ def preprocessing(input):
         input.candidates = [
             candidate
             for candidate in input.candidates
-            if candidate.version.local is not None
-            and "rocm" not in candidate.version.local
+            if candidate.name not in PYTORCH_DISTRIBUTIONS
+            or (
+                candidate.version.local is not None
+                and "rocm" not in candidate.version.local
+            )
         ]
         return input