()
| 185 | |
| 186 | |
| 187 | def get_template_config() -> Config: |
| 188 | config_path = os.path.join(os.path.dirname(__file__), "template", "config.yaml") |
| 189 | config = load_config(config_path) |
| 190 | if ray.is_initialized(): |
| 191 | config.ray_namespace = ray.get_runtime_context().namespace |
| 192 | else: |
| 193 | config.ray_namespace = "trinity_unittest" |
| 194 | return config |
| 195 | |
| 196 | |
| 197 | def get_model_path() -> str: |