.

Author: wangraying

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