use functools.cache unconditionally

rapidsai · Aug 21, 2024 · 11de09e · 11de09e
1 parent 96c6628
commit 11de09e
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/python/cuml/cuml/internals/available_devices.py b/python/cuml/cuml/internals/available_devices.py
@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2022-2023, NVIDIA CORPORATION.
+# Copyright (c) 2022-2024, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,12 +16,8 @@
 from cuml.internals.device_support import GPU_ENABLED
 from cuml.internals.safe_imports import gpu_only_import_from, UnavailableError
 
-try:
-    from functools import cache  # requires Python >= 3.9
-except ImportError:
-    from functools import lru_cache
 
-    cache = lru_cache(maxsize=None)
+from functools import cache
 
 
 def gpu_available_no_context_creation():