(self, version, build_data)
| 36 | """A custom hook to inject TPU dependencies into the core wheel dependencies.""" |
| 37 | |
| 38 | def initialize(self, version, build_data): # pylint: disable=unused-argument |
| 39 | tpu_deps = get_tpu_dependencies() |
| 40 | build_data["dependencies"] = tpu_deps |
| 41 | print(f"Successfully injected {len(tpu_deps)} TPU dependencies into the wheel's core requirements.") |