diff --git a/src/environment.py b/src/environment.py index bbe09ee..dd7cbca 100644 --- a/src/environment.py +++ b/src/environment.py @@ -3,7 +3,7 @@ from random import choices def generate_field() -> List[List[int]]: - return [choices(range(6), weights=[8, 8, 4, 4, 4, 4], k=10) for y in range(10)] + return [choices(range(6), weights=[8, 8, 4, 4, 4, 4], k=10) for _ in range(10)] class Environment: