MCPcopy Create free account
hub / github.com/AI-Hypercomputer/maxtext / initialize

Method initialize

build_hooks.py:38–41  ·  view source on GitHub ↗
(self, version, build_data)

Source from the content-addressed store, hash-verified

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.")

Callers 15

mainFunction · 0.45
mainFunction · 0.45
test_mlp_blockMethod · 0.45
setUpMethod · 0.45
setUpMethod · 0.45
setUpMethod · 0.45
setUpMethod · 0.45
setUpMethod · 0.45
setUpMethod · 0.45
setUpMethod · 0.45

Calls 1

get_tpu_dependenciesFunction · 0.85

Tested by 15

test_mlp_blockMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36
setUpMethod · 0.36