Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions gittensor/cli/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,9 @@
import os
import sys

# Stub heavy imports during shell completion so tab-completion stays fast.
if os.environ.get('_GITT_COMPLETE'):
# Stub heavy imports during shell completion and --help so tab-completion stays
# fast and bittensor's argparse doesn't hijack click's help output.
if os.environ.get('_GITT_COMPLETE') or any(arg in ('-h', '--help') for arg in sys.argv[1:]):
import types as _types

class _Stub(_types.ModuleType):
Expand All @@ -27,7 +28,7 @@ def __getattr__(self, _name):
def __call__(self, *_a, **_kw):
return self

_stub = _Stub('_gitt_completion_stub')
_stub = _Stub('_gitt_cli_stub')
for _pkg in ('bittensor', 'requests'):
sys.modules[_pkg] = _stub

Expand Down
Loading