analyse_datasets_realWorldDRL/config.py at main · learnsyslab/analyse_datasets_realWorldDRL · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
from dataclasses import dataclass, field

@dataclass
class LegoSimpleDiffusion:
    datasets: list = field(default_factory=lambda: [
        "OliverHausdoerfer/stack_lego_simple_diffusion_deploy_1",
        "OliverHausdoerfer/stack_lego_simple_diffusion_deploy_3",
        "OliverHausdoerfer/stack_lego_simple_diffusion_deploy_4"
    ])

@dataclass
class LegoSimpleDitflow:
    datasets: list = field(default_factory=lambda: [
        "OliverHausdoerfer/stack_lego_simple_ditflow_deploy",
        "OliverHausdoerfer/stack_lego_simple_ditflow_deploy_1",
    ])

@dataclass
class LegoSimplePi05:
    datasets: list = field(default_factory=lambda: [
        "OliverHausdoerfer/stack_lego_simple_pi05_deploy_2",
        "OliverHausdoerfer/stack_lego_simple_pi05_deploy_1",
    ])


@dataclass
class SiemensDiffusion:
    datasets: list = field(default_factory=lambda: [
        "OliverHausdoerfer/siemens_diffusion_deploy_1",
    ])

@dataclass
class SiemensDitflow:
    datasets: list = field(default_factory=lambda: [
        "OliverHausdoerfer/siemens_ditflow_deploy_1",
    ])

@dataclass
class SiemensPi05:
    datasets: list = field(default_factory=lambda: [
        "OliverHausdoerfer/siemens_pi05_deploy_1",
    ])

# @dataclass
# class LegoSimpleGeneralizationDiffusion:
#     datasets: list = field(default_factory=lambda: [
#         "OliverHausdoerfer/stack_lego_simple_generalization_diffusion_deploy_1",
#         "OliverHausdoerfer/stack_lego_simple_generalization_diffusion_deploy_2",
#     ])

# @dataclass
# class LegoSimpleGeneralizationDitflow:
#     datasets: list = field(default_factory=lambda: [
#         "OliverHausdoerfer/stack_lego_simple_generalization_ditflow_deploy_2",
#         "OliverHausdoerfer/stack_lego_simple_generalization_ditflow_deploy_1",
#     ])

# @dataclass
# class LegoSimpleGeneralizationPi05:
#     datasets: list = field(default_factory=lambda: [
#         "OliverHausdoerfer/stack_lego_simple_pi05_generalization_deploy_2",
#         "OliverHausdoerfer/stack_lego_simple_pi05_generalization_deploy_1",
#     ])