MCPcopy Create free account
hub / github.com/apple/axlearn / from_flags

Method from_flags

axlearn/cloud/gcp/node_pool_provisioner.py:71–81  ·  view source on GitHub ↗
(cls, fv: flags.FlagValues, **kwargs)

Source from the content-addressed store, hash-verified

69 disk_type: Optional[str] = None
70 boot_disk_kms_key: Optional[str] = None
71 enable_confidential_storage: Optional[bool] = None
72
73 @classmethod
74 def from_flags(cls, fv: flags.FlagValues, **kwargs) -> Config:
75 cfg: NodePoolProvisioner.Config = super().from_flags(fv, **kwargs)
76 cfg.project = gcp_settings("project", fv=fv)
77 cfg.zone = gcp_settings("zone", fv=fv)
78 cfg.cluster = gcp_settings("gke_cluster", fv=fv)
79 cfg.service_account_email = gcp_settings("service_account_email", required=False, fv=fv)
80 cfg.disk_type = gcp_settings("gke_disk_type", required=False, fv=fv)
81 cfg.boot_disk_kms_key = gcp_settings("gke_boot_disk_kms_key", required=False, fv=fv)
82 enable_cs = gcp_settings("gke_enable_confidential_storage", required=False, fv=fv)
83 cfg.enable_confidential_storage = enable_cs.strip().lower() == "true" if enable_cs else None
84 return cfg

Callers

nothing calls this directly

Calls 2

gcp_settingsFunction · 0.90
lowerMethod · 0.80

Tested by

no test coverage detected