equidiff/equi_diffpo/env_runner/base_lowdim_runner.py

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()