diff --git a/libs/experimental/tests/unit_tests/__init__.py b/libs/experimental/tests/unit_tests/__init__.py
index 48e02109f01..0594a6bd14b 100644
--- a/libs/experimental/tests/unit_tests/__init__.py
+++ b/libs/experimental/tests/unit_tests/__init__.py
@@ -1,6 +1,6 @@
 import ctypes
 
-def is_libcublas_available():
+def is_libcublas_available() -> bool:
     try:
         ctypes.CDLL("libcublas.so")
         return True