from typing import Dict from equi_diffpo.policy.base_lowdim_policy import BaseLowdimPolicy class BaseLowdimRunner: def __init__(self, output_dir): self.output_dir = output_dir def run(self, policy: BaseLowdimPolicy) -> Dict: raise NotImplementedError()