MAINT: bimatrix_generators: Define __all__ (#525)

QuantEcon · Feb 6, 2020 · 0e30f47 · 0e30f47
1 parent be16ea5
commit 0e30f47
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/quantecon/game_theory/game_generators/__init__.py b/quantecon/game_theory/game_generators/__init__.py
@@ -3,6 +3,4 @@
 game_theory.game_generators
 
 """
-from .bimatrix_generators import (
-	blotto_game, ranking_game, sgc_game, tournament_game, unit_vector_game
-)
+from .bimatrix_generators import *
diff --git a/quantecon/game_theory/game_generators/bimatrix_generators.py b/quantecon/game_theory/game_generators/bimatrix_generators.py
@@ -99,6 +99,11 @@
 from ...graph_tools import random_tournament_graph
 from ...util.combinatorics import next_k_array, k_array_rank_jit
 
+__all__ = [
+    'blotto_game', 'ranking_game', 'sgc_game', 'tournament_game',
+    'unit_vector_game'
+]
+
 
 def blotto_game(h, t, rho, mu=0, random_state=None):
     """