search

heronsystems / adeptRL

Reinforcement learning framework to accelerate research
GNU General Public License v3.0
204 stars 29 forks source link

[Bug] ObsPreprocessor does not ops name_filters on call #93

Open benbellheron opened 4 years ago

benbellheron commented 4 years ago

Example:

        cpu_ops = [CustomOpFn(name_filters=[dict_key])]
        cpu_preprocessor = ObsPreprocessor(
            cpu_ops,
            Space.from_gym(observation_space),
            Space.dtypes_from_gym(observation_space),
        )

I expect that my CustomOpFn should only be receiving the name_filters specified when calling update_shape, update_dtype and update_obs but it does not occur for update obs.

Relevant file/lines: https://github.com/heronsystems/adeptRL/blob/master/adept/preprocess/observation.py#L51