dodox/Columbus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Always return np.array not th.Tensor

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-08-22 17:23:12 +02:00
parent a41f93beed
commit 0b71d2fe0c
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
columbus/observables.py
View File

@ -337,8 +337,6 @@ class CompositionalObservable(Observable):
def get_observation_space(self): def get_observation_space(self):
num = 0 num = 0
low = 99999
high = -99999
for i, obs in enumerate(self.observables): for i, obs in enumerate(self.observables):
space = obs.get_observation_space() space = obs.get_observation_space()
num += math.prod(space.shape) num += math.prod(space.shape)
@ -352,9 +350,11 @@ class CompositionalObservable(Observable):
shape=(num,), dtype=np.float32) shape=(num,), dtype=np.float32)
def get_observation(self): def get_observation(self):
o = [th.reshape(th.Tensor(obs.get_observation()), (-1,)) o = [obs.get_observation().reshape((-1))
for obs in self.observables] for obs in self.observables]
o = th.hstack(o) o = np.hstack(o)
import pdb
pdb.set_trace()
return o return o
def draw(self): def draw(self):