Add annotations to torchrec/distributed/quant_embeddingbag.py (#1647)

Summary: Pull Request resolved: #1647 Reviewed By: azad-meta Differential Revision: D52957246 fbshipit-source-id: e202943dda3538cc02374f1e27562cb991cf8c16
pytorch · Jan 22, 2024 · ece441d · ece441d
1 parent f2614f9
commit ece441d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrec/distributed/quant_embeddingbag.py b/torchrec/distributed/quant_embeddingbag.py
@@ -65,7 +65,7 @@
     from torch._dynamo import is_compiling as is_torchdynamo_compiling
 except Exception:
 
-    def is_torchdynamo_compiling():  # type: ignore[misc]
+    def is_torchdynamo_compiling() -> bool:  # type: ignore[misc]
         return False