| 161 | ) |
| 162 | |
| 163 | def post_setup(self): |
| 164 | valid_engines = set( |
| 165 | "analytical,analytical-java,interactive,learning,gae,gae-java,gie,gle,glt".split( |
| 166 | "," |
| 167 | ) |
| 168 | ) |
| 169 | for item in [item.strip() for item in self.enabled_engines.split(",")]: |
| 170 | if item not in valid_engines and item != "": |
| 171 | print(f"Not a valid engine name: {item}") |
| 172 | if item == "analytical" or item == "gae": |
| 173 | self.enable_gae = True |
| 174 | if item == "interactive" or item == "gie": |
| 175 | self.enable_gie = True |
| 176 | if item == "graphlearn" or item == "gle": |
| 177 | self.enable_gle = True |
| 178 | if item == "graphlearn-torch" or item == "glt": |
| 179 | self.enable_glt = True |
| 180 | if item == "analytical-java" or item == "gae-java": |
| 181 | self.enable_gae_java = True |
| 182 | |
| 183 | if self.preemptive: |
| 184 | self.gae_resource.requests = None |
| 185 | self.gle_resource.requests = None |
| 186 | self.glt_resource.requests = None |
| 187 | self.gie_executor_resource.requests = None |
| 188 | self.gie_frontend_resource.requests = None |
| 189 | |
| 190 | |
| 191 | @dataclass |