Commit 92b4d3bf authored by Yuxin Wu's avatar Yuxin Wu

fix #860

parent 377bfd0f
......@@ -220,9 +220,8 @@ def get_current_tower_context():
When called inside a TowerContext, returns the TowerContext.
Returns:
a :class:`BaseTowerContext` instance.
a :class:`BaseTowerContext` instance or None, if not called under a TowerContext.
"""
assert _CurrentTowerContext is not None, "The function is supposed to be called under a TowerContext!"
return _CurrentTowerContext
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment