From 0e536ec94b4e5d7ae6d136d896e27a1f441db4ce Mon Sep 17 00:00:00 2001
From: SP Mohanty <spmohanty91@gmail.com>
Date: Thu, 4 Apr 2019 04:36:21 +0200
Subject: [PATCH] minor refactor
---
flatland/core/transitions.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/flatland/core/transitions.py b/flatland/core/transitions.py
index a2bb657..d70f8da 100644
--- a/flatland/core/transitions.py
+++ b/flatland/core/transitions.py
@@ -167,9 +167,11 @@ class GridTransitions(Transitions):
transitions,
allow_diagonal_transitions=False
):
- self.number_of_cell_neighbors = 4
+
if allow_diagonal_transitions:
self.number_of_cell_neighbors = 8
+ else:
+ self.number_of_cell_neighbors = 4
self.transitions = transitions
--
GitLab