.

BaguaSys · Nov 1, 2021 · 5add7f8 · 5add7f8
1 parent dad451b
commit 5add7f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bagua/torch_api/contrib/fuse/optimizer.py b/bagua/torch_api/contrib/fuse/optimizer.py
@@ -1,5 +1,5 @@
 import torch
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, Any
 import copy
 import logging
 from functools import reduce