✅ Upgrade deprecated function

georgian-io · Feb 24, 2023 · 1a4fee3 · 1a4fee3
1 parent e119df3
commit 1a4fee3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -5,7 +5,7 @@
 
 
 def has_gpu():
-    return tf.test.is_gpu_available()
+    return len(tf.config.list_physical_devices('GPU')) > 0
 
 
 MODELS = [