Skip to content

Commit

Permalink
fix: Adjusted errors pointed by check_env
Browse files Browse the repository at this point in the history
  • Loading branch information
RickFqt committed Dec 4, 2024
1 parent 86ecd0c commit 86b4dd5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion urnai/sc2/states/collectables.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,6 +158,7 @@ def build_non_spatial_state(self, obs):
# position 2: number of remaining shards
# self.non_spatial_state[2]=np.count_nonzero(obs.feature_minimap[4]==16)
self.normalize_non_spatial_list()
self.non_spatial_state = np.array(self.non_spatial_state)
return self.non_spatial_state

def calculate_distance(self, x1, y1, x2, y2):
Expand All @@ -175,7 +176,7 @@ def reset(self):
self._state = None
self._dimension = None

def trim_matrix(matrix, x1, y1, x2, y2):
def trim_matrix(self, matrix, x1, y1, x2, y2):
"""
If you have a 2D numpy array
and you want a submatrix of that array,
Expand Down

0 comments on commit 86b4dd5

Please sign in to comment.