File size: 280 Bytes
33c751d
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from typing import Dict
from diffusion_policy.policy.base_image_policy import BaseImagePolicy

class BaseImageRunner:
    def __init__(self, output_dir):
        self.output_dir = output_dir

    def run(self, policy: BaseImagePolicy) -> Dict:
        raise NotImplementedError()