[NFC] polish colossalai/nn/layer/vanilla/__init__.py code style (#1293)

[NFC] polish colossalai/nn/layer/vanilla/init.py code style (#1293)
1000a41f · Liping233 · Frank Lee · 87f679ae · 1000a41f
Commit 1000a41f authored Jul 13, 2022 by Liping233 Committed by Frank Lee Jul 13, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

colossalai/nn/layer/vanilla/__init__.py colossalai/nn/layer/vanilla/__init__.py +3 -4

No files found.
--- a/colossalai/nn/layer/vanilla/__init__.py
+++ b/colossalai/nn/layer/vanilla/__init__.py
-from .layers import (DropPath, VanillaClassifier, VanillaLayerNorm,
+from .layers import (DropPath, VanillaClassifier, VanillaLayerNorm, VanillaPatchEmbedding, WrappedDropout,
-                     VanillaPatchEmbedding, WrappedDropout, WrappedDropPath)
+                     WrappedDropPath)
 __all__ = [
-    "VanillaLayerNorm", "VanillaPatchEmbedding", "VanillaClassifier",
+    "VanillaLayerNorm", "VanillaPatchEmbedding", "VanillaClassifier", "DropPath", "WrappedDropout", "WrappedDropPath"
-    "DropPath", "WrappedDropout", "WrappedDropPath"
 ]