From 52e975a66d28593c29672ef18dd3e0749764d5cb Mon Sep 17 00:00:00 2001 From: genyrosk Date: Fri, 15 Jan 2021 02:09:52 +0100 Subject: [PATCH] make imports avilable directly from the module --- README.md | 2 +- gym_chess/test/v1/test_basic_moves.py | 2 +- gym_chess/test/v1/test_benchmark.py | 2 +- gym_chess/test/v1/test_capture_moves.py | 2 +- gym_chess/test/v1/test_castle_moves.py | 2 +- gym_chess/test/v1/test_king_moves.py | 2 +- gym_chess/test/v1/test_run_moves.py | 2 +- gym_chess/test/v1/test_squares_under_attack.py | 2 +- gym_chess/test/v2/test_basic_moves.py | 2 +- gym_chess/test/v2/test_benchmark.py | 2 +- gym_chess/test/v2/test_capture_moves.py | 2 +- gym_chess/test/v2/test_castle_moves.py | 2 +- gym_chess/test/v2/test_king_moves.py | 2 +- gym_chess/test/v2/test_run_moves.py | 2 +- gym_chess/test/v2/test_squares_under_attack.py | 2 +- 15 files changed, 15 insertions(+), 15 deletions(-) diff --git a/README.md b/README.md index 5e30390..a54164b 100644 --- a/README.md +++ b/README.md @@ -360,7 +360,7 @@ The `v2` environment is over 100 times faster than the `v1` environment. However ```python -from gym_chess.envs import ChessEnvV1, ChessEnvV2 +from gym_chess import ChessEnvV1, ChessEnvV2 env_v1 = ChessEnvV1() env_v2 = ChessEnvV2() diff --git a/gym_chess/test/v1/test_basic_moves.py b/gym_chess/test/v1/test_basic_moves.py index 26f3f46..547ee81 100644 --- a/gym_chess/test/v1/test_basic_moves.py +++ b/gym_chess/test/v1/test_basic_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v1/test_benchmark.py b/gym_chess/test/v1/test_benchmark.py index d0afa31..f7ea010 100644 --- a/gym_chess/test/v1/test_benchmark.py +++ b/gym_chess/test/v1/test_benchmark.py @@ -2,7 +2,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.test.utils import run_test_funcs diff --git a/gym_chess/test/v1/test_capture_moves.py b/gym_chess/test/v1/test_capture_moves.py index 96c116a..e0bfd00 100644 --- a/gym_chess/test/v1/test_capture_moves.py +++ b/gym_chess/test/v1/test_capture_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v1/test_castle_moves.py b/gym_chess/test/v1/test_castle_moves.py index 4624bec..a59b296 100644 --- a/gym_chess/test/v1/test_castle_moves.py +++ b/gym_chess/test/v1/test_castle_moves.py @@ -2,7 +2,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v1/test_king_moves.py b/gym_chess/test/v1/test_king_moves.py index 67624c3..2a1bf18 100644 --- a/gym_chess/test/v1/test_king_moves.py +++ b/gym_chess/test/v1/test_king_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v1/test_run_moves.py b/gym_chess/test/v1/test_run_moves.py index 8a090d9..8769b47 100644 --- a/gym_chess/test/v1/test_run_moves.py +++ b/gym_chess/test/v1/test_run_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v1/test_squares_under_attack.py b/gym_chess/test/v1/test_squares_under_attack.py index 2abcb59..baeadc5 100644 --- a/gym_chess/test/v1/test_squares_under_attack.py +++ b/gym_chess/test/v1/test_squares_under_attack.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV1 +from gym_chess import ChessEnvV1 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v2/test_basic_moves.py b/gym_chess/test/v2/test_basic_moves.py index d3c521c..af083bd 100644 --- a/gym_chess/test/v2/test_basic_moves.py +++ b/gym_chess/test/v2/test_basic_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.envs.chess_v2 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v2/test_benchmark.py b/gym_chess/test/v2/test_benchmark.py index 9fb28bd..cef5a24 100644 --- a/gym_chess/test/v2/test_benchmark.py +++ b/gym_chess/test/v2/test_benchmark.py @@ -2,7 +2,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.test.utils import run_test_funcs diff --git a/gym_chess/test/v2/test_capture_moves.py b/gym_chess/test/v2/test_capture_moves.py index 232a102..5bfc2a8 100644 --- a/gym_chess/test/v2/test_capture_moves.py +++ b/gym_chess/test/v2/test_capture_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v2/test_castle_moves.py b/gym_chess/test/v2/test_castle_moves.py index a31edc4..4d0689e 100644 --- a/gym_chess/test/v2/test_castle_moves.py +++ b/gym_chess/test/v2/test_castle_moves.py @@ -2,7 +2,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.envs.chess_v2 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v2/test_king_moves.py b/gym_chess/test/v2/test_king_moves.py index bf79338..70b7c64 100644 --- a/gym_chess/test/v2/test_king_moves.py +++ b/gym_chess/test/v2/test_king_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.envs.chess_v2 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v2/test_run_moves.py b/gym_chess/test/v2/test_run_moves.py index 5ce37a0..5c1ec91 100644 --- a/gym_chess/test/v2/test_run_moves.py +++ b/gym_chess/test/v2/test_run_moves.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID, diff --git a/gym_chess/test/v2/test_squares_under_attack.py b/gym_chess/test/v2/test_squares_under_attack.py index d865d4b..d9d11f1 100644 --- a/gym_chess/test/v2/test_squares_under_attack.py +++ b/gym_chess/test/v2/test_squares_under_attack.py @@ -1,7 +1,7 @@ from copy import copy import numpy as np -from gym_chess.envs import ChessEnvV2 +from gym_chess import ChessEnvV2 from gym_chess.envs.chess_v1 import ( KING_ID, QUEEN_ID,