code: def main():
      run = wandb.init(entity="mock_server_entity", project="test")
      art = wandb.Artifact("test_artifact", type="model")
      art.add_file("public_collection.py")
      run.link_artifact(art, "mock_server_entity/test/test_port")
      run.finish()
  
      collections = (
          wandb.Api().artifact_type("model", "test").collections()
      )
      assert len(collections) == 2
  
label_desc: Use image node_id 1 for calling a global function with example usage: main() without return types