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

fix #860

parent 377bfd0f
...@@ -220,9 +220,8 @@ def get_current_tower_context(): ...@@ -220,9 +220,8 @@ def get_current_tower_context():
When called inside a TowerContext, returns the TowerContext. When called inside a TowerContext, returns the TowerContext.
Returns: 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 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