#3342 add support for `dir` in wandb plugin initialization Pull request opened by samhita-alla ## Tracking issue ## Why are the changes needed? To enable users to specify

dir

in the

wandb_init

initialization, we can allow a string that refers to the

FlyteDirectory

parameter name. The plugin would then download that directory and pass the local path to the

wandb_init

call. ## What changes were proposed in this pull request? This PR adds support for materializing

dir

in the wandb plugin. @task( container_image=image, secret_requests=[wandb_secret], ) @wandb_init( project=WANDB_PROJECT, entity=WANDB_ENTITY, secret=wandb_secret, dir="run_dir", ) def t1(run_dir: union.FlyteDirectory) -> union.FlyteDirectory: wandb.log({"accuracy": 0.9, "loss": 0.1}) return union.FlyteDirectory(path=wandb.run.dir) ## How was this patch tested? ### Setup process ### Screenshots ## Check all the applicable boxes • I updated the documentation accordingly. • All new and existing tests passed. • All commits are signed-off. ## Related PRs ## Docs link https://bito.ai#summarystart| ### Summary by Bito This pull request introduces a new feature allowing users to specify a 'dir' parameter during wandb plugin initialization. It enhances the plugin's ability to handle FlyteDirectory parameters, improving the flexibility and usability of wandb integration within Flyte tasks. flyteorg/flytekit