[Fix] Restructure for wgit availability as a package (#1018)

facebookresearch · Jun 27, 2022 · 96fcc1c · 96fcc1c
1 parent 064e886
commit 96fcc1c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/fairscale/experimental/__init__.py b/fairscale/experimental/__init__.py
@@ -9,6 +9,6 @@
 
 from typing import List
 
-from . import nn, optim
+from . import nn, optim, wgit
 
 __all__: List[str] = []
diff --git a/fairscale/experimental/wgit/__init__.py b/fairscale/experimental/wgit/__init__.py
@@ -3,7 +3,10 @@
 # This source code is licensed under the BSD license found in the
 # LICENSE file in the root directory of this source tree.
 
+from typing import List
+
 from .repo import Repo
 from .version import __version_tuple__
 
 __version__ = ".".join([str(x) for x in __version_tuple__])
+__all__: List[str] = []