wandb 0.21.0__py3-none-win_amd64.whl → 0.21.2__py3-none-win_amd64.whl

wandb/proto/v6/wandb_telemetry_pb2.py

@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 from wandb.proto import wandb_base_pb2 as wandb_dot_proto_dot_wandb__base__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!wandb/proto/wandb_telemetry.proto\x12\x0ewandb_internal\x1a\x1cwandb/proto/wandb_base.proto\"\xdb\x03\n\x0fTelemetryRecord\x12-\n\x0cimports_init\x18\x01 \x01(\x0b\x32\x17.wandb_internal.Imports\x12/\n\x0eimports_finish\x18\x02 \x01(\x0b\x32\x17.wandb_internal.Imports\x12(\n\x07\x66\x65\x61ture\x18\x03 \x01(\x0b\x32\x17.wandb_internal.Feature\x12\x16\n\x0epython_version\x18\x04 \x01(\t\x12\x13\n\x0b\x63li_version\x18\x05 \x01(\t\x12\x1b\n\x13huggingface_version\x18\x06 \x01(\t\x12 \n\x03\x65nv\x18\x08 \x01(\x0b\x32\x13.wandb_internal.Env\x12%\n\x05label\x18\t \x01(\x0b\x32\x16.wandb_internal.Labels\x12.\n\ndeprecated\x18\n \x01(\x0b\x32\x1a.wandb_internal.Deprecated\x12&\n\x06issues\x18\x0b \x01(\x0b\x32\x16.wandb_internal.Issues\x12\x14\n\x0c\x63ore_version\x18\x0c \x01(\t\x12\x10\n\x08platform\x18\r \x01(\t\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x11\n\x0fTelemetryResult\"\x9a\x0e\n\x07Imports\x12\r\n\x05torch\x18\x01 \x01(\x08\x12\r\n\x05keras\x18\x02 \x01(\x08\x12\x12\n\ntensorflow\x18\x03 \x01(\x08\x12\x0e\n\x06\x66\x61stai\x18\x04 \x01(\x08\x12\x0f\n\x07sklearn\x18\x05 \x01(\x08\x12\x0f\n\x07xgboost\x18\x06 \x01(\x08\x12\x10\n\x08\x63\x61tboost\x18\x07 \x01(\x08\x12\x10\n\x08lightgbm\x18\x08 \x01(\x08\x12\x19\n\x11pytorch_lightning\x18\t \x01(\x08\x12\x0e\n\x06ignite\x18\n \x01(\x08\x12\x14\n\x0ctransformers\x18\x0b \x01(\x08\x12\x0b\n\x03jax\x18\x0c \x01(\x08\x12\x10\n\x08metaflow\x18\r \x01(\x08\x12\x10\n\x08\x61llennlp\x18\x0e \x01(\x08\x12\x11\n\tautogluon\x18\x0f \x01(\x08\x12\x11\n\tautokeras\x18\x10 \x01(\x08\x12\x10\n\x08\x63\x61talyst\x18\x12 \x01(\x08\x12\x10\n\x08\x64\x65\x65pchem\x18\x15 \x01(\x08\x12\x0f\n\x07\x64\x65\x65pctr\x18\x16 \x01(\x08\x12\x0f\n\x07pycaret\x18\x1c \x01(\x08\x12\x14\n\x0cpytorchvideo\x18\x1d \x01(\x08\x12\x0b\n\x03ray\x18\x1e \x01(\x08\x12\x1a\n\x12simpletransformers\x18\x1f \x01(\x08\x12\x0e\n\x06skorch\x18  \x01(\x08\x12\r\n\x05spacy\x18! \x01(\x08\x12\r\n\x05\x66lash\x18\" \x01(\x08\x12\x0e\n\x06optuna\x18# \x01(\x08\x12\x0f\n\x07recbole\x18$ \x01(\x08\x12\x0c\n\x04mmcv\x18% \x01(\x08\x12\r\n\x05mmdet\x18& \x01(\x08\x12\x11\n\ttorchdrug\x18\' \x01(\x08\x12\x11\n\ttorchtext\x18( \x01(\x08\x12\x13\n\x0btorchvision\x18) \x01(\x08\x12\r\n\x05\x65legy\x18* \x01(\x08\x12\x12\n\ndetectron2\x18+ \x01(\x08\x12\r\n\x05\x66lair\x18, \x01(\x08\x12\x0c\n\x04\x66lax\x18- \x01(\x08\x12\x0c\n\x04syft\x18. \x01(\x08\x12\x0b\n\x03TTS\x18/ \x01(\x08\x12\r\n\x05monai\x18\x30 \x01(\x08\x12\x17\n\x0fhuggingface_hub\x18\x31 \x01(\x08\x12\r\n\x05hydra\x18\x32 \x01(\x08\x12\x10\n\x08\x64\x61tasets\x18\x33 \x01(\x08\x12\x0e\n\x06sacred\x18\x34 \x01(\x08\x12\x0e\n\x06joblib\x18\x35 \x01(\x08\x12\x0c\n\x04\x64\x61sk\x18\x36 \x01(\x08\x12\x11\n\tpaddleocr\x18\x38 \x01(\x08\x12\r\n\x05ppdet\x18\x39 \x01(\x08\x12\x11\n\tpaddleseg\x18: \x01(\x08\x12\x11\n\tpaddlenlp\x18; \x01(\x08\x12\r\n\x05mmseg\x18< \x01(\x08\x12\r\n\x05mmocr\x18= \x01(\x08\x12\r\n\x05mmcls\x18> \x01(\x08\x12\x0c\n\x04timm\x18? \x01(\x08\x12\x0f\n\x07\x66\x61irseq\x18@ \x01(\x08\x12\x12\n\ndeepchecks\x18\x41 \x01(\x08\x12\x10\n\x08\x63omposer\x18\x42 \x01(\x08\x12\x10\n\x08sparseml\x18\x43 \x01(\x08\x12\x10\n\x08\x61nomalib\x18\x44 \x01(\x08\x12\r\n\x05zenml\x18\x45 \x01(\x08\x12\x12\n\ncolossalai\x18\x46 \x01(\x08\x12\x12\n\naccelerate\x18G \x01(\x08\x12\x0e\n\x06merlin\x18H \x01(\x08\x12\x0f\n\x07nanodet\x18I \x01(\x08\x12#\n\x1bsegmentation_models_pytorch\x18J \x01(\x08\x12\x1d\n\x15sentence_transformers\x18K \x01(\x08\x12\x0b\n\x03\x64gl\x18L \x01(\x08\x12\x17\n\x0ftorch_geometric\x18M \x01(\x08\x12\x0c\n\x04jina\x18N \x01(\x08\x12\x0e\n\x06kornia\x18O \x01(\x08\x12\x16\n\x0e\x61lbumentations\x18P \x01(\x08\x12\x10\n\x08keras_cv\x18Q \x01(\x08\x12\x10\n\x08mmengine\x18R \x01(\x08\x12\x11\n\tdiffusers\x18S \x01(\x08\x12\x0b\n\x03trl\x18T \x01(\x08\x12\x0c\n\x04trlx\x18U \x01(\x08\x12\x11\n\tlangchain\x18V \x01(\x08\x12\x13\n\x0bllama_index\x18W \x01(\x08\x12\x15\n\rstability_sdk\x18X \x01(\x08\x12\x0f\n\x07prefect\x18Y \x01(\x08\x12\x13\n\x0bprefect_ray\x18Z \x01(\x08\x12\x10\n\x08pinecone\x18[ \x01(\x08\x12\x10\n\x08\x63hromadb\x18\\ \x01(\x08\x12\x10\n\x08weaviate\x18] \x01(\x08\x12\x13\n\x0bpromptlayer\x18^ \x01(\x08\x12\x0e\n\x06openai\x18_ \x01(\x08\x12\x0e\n\x06\x63ohere\x18` \x01(\x08\x12\x11\n\tanthropic\x18\x61 \x01(\x08\x12\x0c\n\x04peft\x18\x62 \x01(\x08\x12\x0f\n\x07optimum\x18\x63 \x01(\x08\x12\x10\n\x08\x65valuate\x18\x64 \x01(\x08\x12\x10\n\x08langflow\x18\x65 \x01(\x08\x12\x12\n\nkeras_core\x18\x66 \x01(\x08\x12\x18\n\x10lightning_fabric\x18g \x01(\x08\x12\x1c\n\x14\x63urated_transformers\x18h \x01(\x08\x12\x0e\n\x06orjson\x18i \x01(\x08\x12\x11\n\tlightning\x18j \x01(\x08J\x04\x08\x11\x10\x12J\x04\x08\x13\x10\x14J\x04\x08\x14\x10\x15J\x04\x08\x17\x10\x18J\x04\x08\x18\x10\x19J\x04\x08\x19\x10\x1aJ\x04\x08\x1a\x10\x1bJ\x04\x08\x1b\x10\x1cJ\x04\x08\x37\x10\x38\"\x95\r\n\x07\x46\x65\x61ture\x12\r\n\x05watch\x18\x01 \x01(\x08\x12\x0e\n\x06\x66inish\x18\x02 \x01(\x08\x12\x0c\n\x04save\x18\x03 \x01(\x08\x12\x0f\n\x07offline\x18\x04 \x01(\x08\x12\x0f\n\x07resumed\x18\x05 \x01(\x08\x12\x0c\n\x04grpc\x18\x06 \x01(\x08\x12\x0e\n\x06metric\x18\x07 \x01(\x08\x12\r\n\x05keras\x18\x08 \x01(\x08\x12\x11\n\tsagemaker\x18\t \x01(\x08\x12\x1c\n\x14\x61rtifact_incremental\x18\n \x01(\x08\x12\x10\n\x08metaflow\x18\x0b \x01(\x08\x12\x0f\n\x07prodigy\x18\x0c \x01(\x08\x12\x15\n\rset_init_name\x18\r \x01(\x08\x12\x13\n\x0bset_init_id\x18\x0e \x01(\x08\x12\x15\n\rset_init_tags\x18\x0f \x01(\x08\x12\x17\n\x0fset_init_config\x18\x10 \x01(\x08\x12\x14\n\x0cset_run_name\x18\x11 \x01(\x08\x12\x14\n\x0cset_run_tags\x18\x12 \x01(\x08\x12\x17\n\x0fset_config_item\x18\x13 \x01(\x08\x12\x0e\n\x06launch\x18\x14 \x01(\x08\x12\x1c\n\x14torch_profiler_trace\x18\x15 \x01(\x08\x12\x0b\n\x03sb3\x18\x16 \x01(\x08\x12\x17\n\x0finit_return_run\x18\x18 \x01(\x08\x12\x1f\n\x17lightgbm_wandb_callback\x18\x19 \x01(\x08\x12\x1c\n\x14lightgbm_log_summary\x18\x1a \x01(\x08\x12\x1f\n\x17\x63\x61tboost_wandb_callback\x18\x1b \x01(\x08\x12\x1c\n\x14\x63\x61tboost_log_summary\x18\x1c \x01(\x08\x12\x17\n\x0ftensorboard_log\x18\x1d \x01(\x08\x12\x16\n\x0e\x65stimator_hook\x18\x1e \x01(\x08\x12\x1e\n\x16xgboost_wandb_callback\x18\x1f \x01(\x08\x12\"\n\x1axgboost_old_wandb_callback\x18  \x01(\x08\x12\x0e\n\x06\x61ttach\x18! \x01(\x08\x12\x19\n\x11tensorboard_patch\x18\" \x01(\x08\x12\x18\n\x10tensorboard_sync\x18# \x01(\x08\x12\x15\n\rkfp_wandb_log\x18$ \x01(\x08\x12\x1b\n\x13maybe_run_overwrite\x18% \x01(\x08\x12\x1c\n\x14keras_metrics_logger\x18& \x01(\x08\x12\x1e\n\x16keras_model_checkpoint\x18\' \x01(\x08\x12!\n\x19keras_wandb_eval_callback\x18( \x01(\x08\x12\x1d\n\x15\x66low_control_overflow\x18) \x01(\x08\x12\x0c\n\x04sync\x18* \x01(\x08\x12\x1d\n\x15\x66low_control_disabled\x18+ \x01(\x08\x12\x1b\n\x13\x66low_control_custom\x18, \x01(\x08\x12\x14\n\x0copen_metrics\x18. \x01(\x08\x12\x1a\n\x12ultralytics_yolov8\x18/ \x01(\x08\x12\x17\n\x0fimporter_mlflow\x18\x30 \x01(\x08\x12\x15\n\rsync_tfevents\x18\x31 \x01(\x08\x12\x15\n\rasync_uploads\x18\x32 \x01(\x08\x12\x16\n\x0eopenai_autolog\x18\x33 \x01(\x08\x12\x18\n\x10langchain_tracer\x18\x34 \x01(\x08\x12\x16\n\x0e\x63ohere_autolog\x18\x35 \x01(\x08\x12\x1b\n\x13hf_pipeline_autolog\x18\x36 \x01(\x08\x12\r\n\x05lib_c\x18\x38 \x01(\x08\x12\x0f\n\x07lib_cpp\x18\x39 \x01(\x08\x12\x19\n\x11openai_finetuning\x18: \x01(\x08\x12\x19\n\x11\x64iffusers_autolog\x18; \x01(\x08\x12\x1f\n\x17lightning_fabric_logger\x18< \x01(\x08\x12\x14\n\x0cset_step_log\x18= \x01(\x08\x12\x13\n\x0bset_summary\x18> \x01(\x08\x12\x16\n\x0emetric_summary\x18? \x01(\x08\x12\x13\n\x0bmetric_goal\x18@ \x01(\x08\x12\x15\n\rmetric_hidden\x18\x41 \x01(\x08\x12\x18\n\x10metric_step_sync\x18\x42 \x01(\x08\x12\x13\n\x0bshared_mode\x18\x43 \x01(\x08\x12#\n\x1bserver_side_derived_summary\x18\x44 \x01(\x08\x12\x1b\n\x13user_provided_label\x18\x45 \x01(\x08\x12\x1e\n\x16\x64\x63gm_profiling_enabled\x18\x46 \x01(\x08\x12\x11\n\tfork_mode\x18G \x01(\x08\x12\x13\n\x0brewind_mode\x18H \x01(\x08J\x04\x08\x17\x10\x18J\x04\x08-\x10.J\x04\x08\x37\x10\x38\"\xc9\x01\n\x03\x45nv\x12\x0f\n\x07jupyter\x18\x01 \x01(\x08\x12\x0e\n\x06kaggle\x18\x02 \x01(\x08\x12\x0f\n\x07windows\x18\x03 \x01(\x08\x12\x0e\n\x06m1_gpu\x18\x04 \x01(\x08\x12\x10\n\x08trainium\x18\n \x01(\x08\x12\x0b\n\x03pex\x18\x0b \x01(\x08\x12\r\n\x05\x63olab\x18\x0c \x01(\x08\x12\x0f\n\x07ipython\x18\r \x01(\x08\x12\x12\n\naws_lambda\x18\x0e \x01(\x08\x12\x0f\n\x07\x61md_gpu\x18\x0f \x01(\x08J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\n\"H\n\x06Labels\x12\x13\n\x0b\x63ode_string\x18\x01 \x01(\t\x12\x13\n\x0brepo_string\x18\x02 \x01(\t\x12\x14\n\x0c\x63ode_version\x18\x03 \x01(\t\"\xd5\x05\n\nDeprecated\x12!\n\x19keras_callback__data_type\x18\x01 \x01(\x08\x12\r\n\x05plots\x18\x05 \x01(\x08\x12!\n\x19init__config_include_keys\x18\x07 \x01(\x08\x12!\n\x19init__config_exclude_keys\x18\x08 \x01(\x08\x12\"\n\x1akeras_callback__save_model\x18\t \x01(\x08\x12\x18\n\x10langchain_tracer\x18\n \x01(\x08\x12\x1a\n\x12\x61rtifact__get_path\x18\x0b \x01(\x08\x12#\n\x1b\x61rtifactmanifestentry__name\x18\x0c \x01(\x08\x12\x1e\n\x16\x61pi__artifact_versions\x18\r \x01(\x08\x12(\n artifact_collection__change_type\x18\x0e \x01(\x08\x12\x1f\n\x17run__define_metric_copy\x18\x0f \x01(\x08\x12\x14\n\x0crun_disabled\x18\x10 \x01(\x08\x12\x16\n\x0ekeras_callback\x18\x11 \x01(\x08\x12$\n\x1crun__define_metric_best_goal\x18\x12 \x01(\x08\x12\x19\n\x11run__finish_quiet\x18\x13 \x01(\x08\x12\x18\n\x10run__reinit_bool\x18\x14 \x01(\x08\x12\x14\n\x0crun__get_url\x18\x15 \x01(\x08\x12\x19\n\x11run__project_name\x18\x16 \x01(\x08\x12\x1c\n\x14run__get_project_url\x18\x17 \x01(\x08\x12\x1a\n\x12run__get_sweep_url\x18\x18 \x01(\x08\x12 \n\x18run__use_artifact_use_as\x18\x19 \x01(\x08\x12\x18\n\x10\x61rtifact__use_as\x18\x1a \x01(\x08\x12\x1d\n\x15\x61rtifact__init_use_as\x18\x1b \x01(\x08J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x06\x10\x07\"|\n\x06Issues\x12%\n\x1dsettings__validation_warnings\x18\x01 \x01(\x08\x12!\n\x19settings__unexpected_args\x18\x02 \x01(\x08\x12(\n settings__preprocessing_warnings\x18\x03 \x01(\x08\x42\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!wandb/proto/wandb_telemetry.proto\x12\x0ewandb_internal\x1a\x1cwandb/proto/wandb_base.proto\"\xdb\x03\n\x0fTelemetryRecord\x12-\n\x0cimports_init\x18\x01 \x01(\x0b\x32\x17.wandb_internal.Imports\x12/\n\x0eimports_finish\x18\x02 \x01(\x0b\x32\x17.wandb_internal.Imports\x12(\n\x07\x66\x65\x61ture\x18\x03 \x01(\x0b\x32\x17.wandb_internal.Feature\x12\x16\n\x0epython_version\x18\x04 \x01(\t\x12\x13\n\x0b\x63li_version\x18\x05 \x01(\t\x12\x1b\n\x13huggingface_version\x18\x06 \x01(\t\x12 \n\x03\x65nv\x18\x08 \x01(\x0b\x32\x13.wandb_internal.Env\x12%\n\x05label\x18\t \x01(\x0b\x32\x16.wandb_internal.Labels\x12.\n\ndeprecated\x18\n \x01(\x0b\x32\x1a.wandb_internal.Deprecated\x12&\n\x06issues\x18\x0b \x01(\x0b\x32\x16.wandb_internal.Issues\x12\x14\n\x0c\x63ore_version\x18\x0c \x01(\t\x12\x10\n\x08platform\x18\r \x01(\t\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x11\n\x0fTelemetryResult\"\x9a\x0e\n\x07Imports\x12\r\n\x05torch\x18\x01 \x01(\x08\x12\r\n\x05keras\x18\x02 \x01(\x08\x12\x12\n\ntensorflow\x18\x03 \x01(\x08\x12\x0e\n\x06\x66\x61stai\x18\x04 \x01(\x08\x12\x0f\n\x07sklearn\x18\x05 \x01(\x08\x12\x0f\n\x07xgboost\x18\x06 \x01(\x08\x12\x10\n\x08\x63\x61tboost\x18\x07 \x01(\x08\x12\x10\n\x08lightgbm\x18\x08 \x01(\x08\x12\x19\n\x11pytorch_lightning\x18\t \x01(\x08\x12\x0e\n\x06ignite\x18\n \x01(\x08\x12\x14\n\x0ctransformers\x18\x0b \x01(\x08\x12\x0b\n\x03jax\x18\x0c \x01(\x08\x12\x10\n\x08metaflow\x18\r \x01(\x08\x12\x10\n\x08\x61llennlp\x18\x0e \x01(\x08\x12\x11\n\tautogluon\x18\x0f \x01(\x08\x12\x11\n\tautokeras\x18\x10 \x01(\x08\x12\x10\n\x08\x63\x61talyst\x18\x12 \x01(\x08\x12\x10\n\x08\x64\x65\x65pchem\x18\x15 \x01(\x08\x12\x0f\n\x07\x64\x65\x65pctr\x18\x16 \x01(\x08\x12\x0f\n\x07pycaret\x18\x1c \x01(\x08\x12\x14\n\x0cpytorchvideo\x18\x1d \x01(\x08\x12\x0b\n\x03ray\x18\x1e \x01(\x08\x12\x1a\n\x12simpletransformers\x18\x1f \x01(\x08\x12\x0e\n\x06skorch\x18  \x01(\x08\x12\r\n\x05spacy\x18! \x01(\x08\x12\r\n\x05\x66lash\x18\" \x01(\x08\x12\x0e\n\x06optuna\x18# \x01(\x08\x12\x0f\n\x07recbole\x18$ \x01(\x08\x12\x0c\n\x04mmcv\x18% \x01(\x08\x12\r\n\x05mmdet\x18& \x01(\x08\x12\x11\n\ttorchdrug\x18\' \x01(\x08\x12\x11\n\ttorchtext\x18( \x01(\x08\x12\x13\n\x0btorchvision\x18) \x01(\x08\x12\r\n\x05\x65legy\x18* \x01(\x08\x12\x12\n\ndetectron2\x18+ \x01(\x08\x12\r\n\x05\x66lair\x18, \x01(\x08\x12\x0c\n\x04\x66lax\x18- \x01(\x08\x12\x0c\n\x04syft\x18. \x01(\x08\x12\x0b\n\x03TTS\x18/ \x01(\x08\x12\r\n\x05monai\x18\x30 \x01(\x08\x12\x17\n\x0fhuggingface_hub\x18\x31 \x01(\x08\x12\r\n\x05hydra\x18\x32 \x01(\x08\x12\x10\n\x08\x64\x61tasets\x18\x33 \x01(\x08\x12\x0e\n\x06sacred\x18\x34 \x01(\x08\x12\x0e\n\x06joblib\x18\x35 \x01(\x08\x12\x0c\n\x04\x64\x61sk\x18\x36 \x01(\x08\x12\x11\n\tpaddleocr\x18\x38 \x01(\x08\x12\r\n\x05ppdet\x18\x39 \x01(\x08\x12\x11\n\tpaddleseg\x18: \x01(\x08\x12\x11\n\tpaddlenlp\x18; \x01(\x08\x12\r\n\x05mmseg\x18< \x01(\x08\x12\r\n\x05mmocr\x18= \x01(\x08\x12\r\n\x05mmcls\x18> \x01(\x08\x12\x0c\n\x04timm\x18? \x01(\x08\x12\x0f\n\x07\x66\x61irseq\x18@ \x01(\x08\x12\x12\n\ndeepchecks\x18\x41 \x01(\x08\x12\x10\n\x08\x63omposer\x18\x42 \x01(\x08\x12\x10\n\x08sparseml\x18\x43 \x01(\x08\x12\x10\n\x08\x61nomalib\x18\x44 \x01(\x08\x12\r\n\x05zenml\x18\x45 \x01(\x08\x12\x12\n\ncolossalai\x18\x46 \x01(\x08\x12\x12\n\naccelerate\x18G \x01(\x08\x12\x0e\n\x06merlin\x18H \x01(\x08\x12\x0f\n\x07nanodet\x18I \x01(\x08\x12#\n\x1bsegmentation_models_pytorch\x18J \x01(\x08\x12\x1d\n\x15sentence_transformers\x18K \x01(\x08\x12\x0b\n\x03\x64gl\x18L \x01(\x08\x12\x17\n\x0ftorch_geometric\x18M \x01(\x08\x12\x0c\n\x04jina\x18N \x01(\x08\x12\x0e\n\x06kornia\x18O \x01(\x08\x12\x16\n\x0e\x61lbumentations\x18P \x01(\x08\x12\x10\n\x08keras_cv\x18Q \x01(\x08\x12\x10\n\x08mmengine\x18R \x01(\x08\x12\x11\n\tdiffusers\x18S \x01(\x08\x12\x0b\n\x03trl\x18T \x01(\x08\x12\x0c\n\x04trlx\x18U \x01(\x08\x12\x11\n\tlangchain\x18V \x01(\x08\x12\x13\n\x0bllama_index\x18W \x01(\x08\x12\x15\n\rstability_sdk\x18X \x01(\x08\x12\x0f\n\x07prefect\x18Y \x01(\x08\x12\x13\n\x0bprefect_ray\x18Z \x01(\x08\x12\x10\n\x08pinecone\x18[ \x01(\x08\x12\x10\n\x08\x63hromadb\x18\\ \x01(\x08\x12\x10\n\x08weaviate\x18] \x01(\x08\x12\x13\n\x0bpromptlayer\x18^ \x01(\x08\x12\x0e\n\x06openai\x18_ \x01(\x08\x12\x0e\n\x06\x63ohere\x18` \x01(\x08\x12\x11\n\tanthropic\x18\x61 \x01(\x08\x12\x0c\n\x04peft\x18\x62 \x01(\x08\x12\x0f\n\x07optimum\x18\x63 \x01(\x08\x12\x10\n\x08\x65valuate\x18\x64 \x01(\x08\x12\x10\n\x08langflow\x18\x65 \x01(\x08\x12\x12\n\nkeras_core\x18\x66 \x01(\x08\x12\x18\n\x10lightning_fabric\x18g \x01(\x08\x12\x1c\n\x14\x63urated_transformers\x18h \x01(\x08\x12\x0e\n\x06orjson\x18i \x01(\x08\x12\x11\n\tlightning\x18j \x01(\x08J\x04\x08\x11\x10\x12J\x04\x08\x13\x10\x14J\x04\x08\x14\x10\x15J\x04\x08\x17\x10\x18J\x04\x08\x18\x10\x19J\x04\x08\x19\x10\x1aJ\x04\x08\x1a\x10\x1bJ\x04\x08\x1b\x10\x1cJ\x04\x08\x37\x10\x38\"\x95\r\n\x07\x46\x65\x61ture\x12\r\n\x05watch\x18\x01 \x01(\x08\x12\x0e\n\x06\x66inish\x18\x02 \x01(\x08\x12\x0c\n\x04save\x18\x03 \x01(\x08\x12\x0f\n\x07offline\x18\x04 \x01(\x08\x12\x0f\n\x07resumed\x18\x05 \x01(\x08\x12\x0c\n\x04grpc\x18\x06 \x01(\x08\x12\x0e\n\x06metric\x18\x07 \x01(\x08\x12\r\n\x05keras\x18\x08 \x01(\x08\x12\x11\n\tsagemaker\x18\t \x01(\x08\x12\x1c\n\x14\x61rtifact_incremental\x18\n \x01(\x08\x12\x10\n\x08metaflow\x18\x0b \x01(\x08\x12\x0f\n\x07prodigy\x18\x0c \x01(\x08\x12\x15\n\rset_init_name\x18\r \x01(\x08\x12\x13\n\x0bset_init_id\x18\x0e \x01(\x08\x12\x15\n\rset_init_tags\x18\x0f \x01(\x08\x12\x17\n\x0fset_init_config\x18\x10 \x01(\x08\x12\x14\n\x0cset_run_name\x18\x11 \x01(\x08\x12\x14\n\x0cset_run_tags\x18\x12 \x01(\x08\x12\x17\n\x0fset_config_item\x18\x13 \x01(\x08\x12\x0e\n\x06launch\x18\x14 \x01(\x08\x12\x1c\n\x14torch_profiler_trace\x18\x15 \x01(\x08\x12\x0b\n\x03sb3\x18\x16 \x01(\x08\x12\x17\n\x0finit_return_run\x18\x18 \x01(\x08\x12\x1f\n\x17lightgbm_wandb_callback\x18\x19 \x01(\x08\x12\x1c\n\x14lightgbm_log_summary\x18\x1a \x01(\x08\x12\x1f\n\x17\x63\x61tboost_wandb_callback\x18\x1b \x01(\x08\x12\x1c\n\x14\x63\x61tboost_log_summary\x18\x1c \x01(\x08\x12\x17\n\x0ftensorboard_log\x18\x1d \x01(\x08\x12\x16\n\x0e\x65stimator_hook\x18\x1e \x01(\x08\x12\x1e\n\x16xgboost_wandb_callback\x18\x1f \x01(\x08\x12\"\n\x1axgboost_old_wandb_callback\x18  \x01(\x08\x12\x0e\n\x06\x61ttach\x18! \x01(\x08\x12\x19\n\x11tensorboard_patch\x18\" \x01(\x08\x12\x18\n\x10tensorboard_sync\x18# \x01(\x08\x12\x15\n\rkfp_wandb_log\x18$ \x01(\x08\x12\x1b\n\x13maybe_run_overwrite\x18% \x01(\x08\x12\x1c\n\x14keras_metrics_logger\x18& \x01(\x08\x12\x1e\n\x16keras_model_checkpoint\x18\' \x01(\x08\x12!\n\x19keras_wandb_eval_callback\x18( \x01(\x08\x12\x1d\n\x15\x66low_control_overflow\x18) \x01(\x08\x12\x0c\n\x04sync\x18* \x01(\x08\x12\x1d\n\x15\x66low_control_disabled\x18+ \x01(\x08\x12\x1b\n\x13\x66low_control_custom\x18, \x01(\x08\x12\x14\n\x0copen_metrics\x18. \x01(\x08\x12\x1a\n\x12ultralytics_yolov8\x18/ \x01(\x08\x12\x17\n\x0fimporter_mlflow\x18\x30 \x01(\x08\x12\x15\n\rsync_tfevents\x18\x31 \x01(\x08\x12\x15\n\rasync_uploads\x18\x32 \x01(\x08\x12\x16\n\x0eopenai_autolog\x18\x33 \x01(\x08\x12\x18\n\x10langchain_tracer\x18\x34 \x01(\x08\x12\x16\n\x0e\x63ohere_autolog\x18\x35 \x01(\x08\x12\x1b\n\x13hf_pipeline_autolog\x18\x36 \x01(\x08\x12\r\n\x05lib_c\x18\x38 \x01(\x08\x12\x0f\n\x07lib_cpp\x18\x39 \x01(\x08\x12\x19\n\x11openai_finetuning\x18: \x01(\x08\x12\x19\n\x11\x64iffusers_autolog\x18; \x01(\x08\x12\x1f\n\x17lightning_fabric_logger\x18< \x01(\x08\x12\x14\n\x0cset_step_log\x18= \x01(\x08\x12\x13\n\x0bset_summary\x18> \x01(\x08\x12\x16\n\x0emetric_summary\x18? \x01(\x08\x12\x13\n\x0bmetric_goal\x18@ \x01(\x08\x12\x15\n\rmetric_hidden\x18\x41 \x01(\x08\x12\x18\n\x10metric_step_sync\x18\x42 \x01(\x08\x12\x13\n\x0bshared_mode\x18\x43 \x01(\x08\x12#\n\x1bserver_side_derived_summary\x18\x44 \x01(\x08\x12\x1b\n\x13user_provided_label\x18\x45 \x01(\x08\x12\x1e\n\x16\x64\x63gm_profiling_enabled\x18\x46 \x01(\x08\x12\x11\n\tfork_mode\x18G \x01(\x08\x12\x13\n\x0brewind_mode\x18H \x01(\x08J\x04\x08\x17\x10\x18J\x04\x08-\x10.J\x04\x08\x37\x10\x38\"\xc9\x01\n\x03\x45nv\x12\x0f\n\x07jupyter\x18\x01 \x01(\x08\x12\x0e\n\x06kaggle\x18\x02 \x01(\x08\x12\x0f\n\x07windows\x18\x03 \x01(\x08\x12\x0e\n\x06m1_gpu\x18\x04 \x01(\x08\x12\x10\n\x08trainium\x18\n \x01(\x08\x12\x0b\n\x03pex\x18\x0b \x01(\x08\x12\r\n\x05\x63olab\x18\x0c \x01(\x08\x12\x0f\n\x07ipython\x18\r \x01(\x08\x12\x12\n\naws_lambda\x18\x0e \x01(\x08\x12\x0f\n\x07\x61md_gpu\x18\x0f \x01(\x08J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\n\"H\n\x06Labels\x12\x13\n\x0b\x63ode_string\x18\x01 \x01(\t\x12\x13\n\x0brepo_string\x18\x02 \x01(\t\x12\x14\n\x0c\x63ode_version\x18\x03 \x01(\t\"\xc2\x06\n\nDeprecated\x12!\n\x19keras_callback__data_type\x18\x01 \x01(\x08\x12\r\n\x05plots\x18\x05 \x01(\x08\x12!\n\x19init__config_include_keys\x18\x07 \x01(\x08\x12!\n\x19init__config_exclude_keys\x18\x08 \x01(\x08\x12\"\n\x1akeras_callback__save_model\x18\t \x01(\x08\x12\x18\n\x10langchain_tracer\x18\n \x01(\x08\x12\x1a\n\x12\x61rtifact__get_path\x18\x0b \x01(\x08\x12#\n\x1b\x61rtifactmanifestentry__name\x18\x0c \x01(\x08\x12\x1e\n\x16\x61pi__artifact_versions\x18\r \x01(\x08\x12(\n artifact_collection__change_type\x18\x0e \x01(\x08\x12\x1f\n\x17run__define_metric_copy\x18\x0f \x01(\x08\x12\x14\n\x0crun_disabled\x18\x10 \x01(\x08\x12\x16\n\x0ekeras_callback\x18\x11 \x01(\x08\x12$\n\x1crun__define_metric_best_goal\x18\x12 \x01(\x08\x12\x19\n\x11run__finish_quiet\x18\x13 \x01(\x08\x12\x18\n\x10run__reinit_bool\x18\x14 \x01(\x08\x12\x14\n\x0crun__get_url\x18\x15 \x01(\x08\x12\x19\n\x11run__project_name\x18\x16 \x01(\x08\x12\x1c\n\x14run__get_project_url\x18\x17 \x01(\x08\x12\x1a\n\x12run__get_sweep_url\x18\x18 \x01(\x08\x12 \n\x18run__use_artifact_use_as\x18\x19 \x01(\x08\x12\x18\n\x10\x61rtifact__use_as\x18\x1a \x01(\x08\x12\x1d\n\x15\x61rtifact__init_use_as\x18\x1b \x01(\x08\x12\"\n\x1a\x62\x65ta__workflows__log_model\x18\x1c \x01(\x08\x12\"\n\x1a\x62\x65ta__workflows__use_model\x18\x1d \x01(\x08\x12#\n\x1b\x62\x65ta__workflows__link_model\x18\x1e \x01(\x08J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x06\x10\x07\"|\n\x06Issues\x12%\n\x1dsettings__validation_warnings\x18\x01 \x01(\x08\x12!\n\x19settings__unexpected_args\x18\x02 \x01(\x08\x12(\n settings__preprocessing_warnings\x18\x03 \x01(\x08\x42\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -46,7 +46,7 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_LABELS']._serialized_start=4293
   _globals['_LABELS']._serialized_end=4365
   _globals['_DEPRECATED']._serialized_start=4368
-  _globals['_DEPRECATED']._serialized_end=5093
-  _globals['_ISSUES']._serialized_start=5095
-  _globals['_ISSUES']._serialized_end=5219
+  _globals['_DEPRECATED']._serialized_end=5202
+  _globals['_ISSUES']._serialized_start=5204
+  _globals['_ISSUES']._serialized_end=5328
 # @@protoc_insertion_point(module_scope)

wandb/proto/wandb_deprecated.py

@@ -25,6 +25,9 @@ DEPRECATED_FEATURES = Literal[
     "run__use_artifact_use_as",
     "artifact__use_as",
     "artifact__init_use_as",
+    "beta__workflows__log_model",
+    "beta__workflows__use_model",
+    "beta__workflows__link_model",
 ]
 
 class Deprecated:
@@ -51,3 +54,6 @@ class Deprecated:
     run__use_artifact_use_as: DEPRECATED_FEATURES = "run__use_artifact_use_as"
     artifact__use_as: DEPRECATED_FEATURES = "artifact__use_as"
     artifact__init_use_as: DEPRECATED_FEATURES = "artifact__init_use_as"
+    beta__workflows__log_model: DEPRECATED_FEATURES = "beta__workflows__log_model"
+    beta__workflows__use_model: DEPRECATED_FEATURES = "beta__workflows__use_model"
+    beta__workflows__link_model: DEPRECATED_FEATURES = "beta__workflows__link_model"

wandb/sdk/artifacts/_factories.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from wandb import env
+
+from .storage_layout import StorageLayout
+from .storage_policies import WandbStoragePolicy
+
+if TYPE_CHECKING:
+    from .storage_policy import StoragePolicy
+
+
+def make_storage_policy() -> StoragePolicy:
+    """A factory function that returns the default StoragePolicy for the current environment."""
+    layout = StorageLayout.V1 if env.get_use_v1_artifacts() else StorageLayout.V2
+    return WandbStoragePolicy.from_config({"storageLayout": layout})

wandb/sdk/artifacts/_generated/__init__.py

@@ -1,24 +1,63 @@
 # Generated by ariadne-codegen
 
 from .add_aliases import AddAliases, AddAliasesAddAliases
+from .artifact_by_id import ArtifactByID
+from .artifact_by_name import ArtifactByName, ArtifactByNameProject
+from .artifact_collection_membership_file_urls import (
+    ArtifactCollectionMembershipFileUrls,
+    ArtifactCollectionMembershipFileUrlsProject,
+    ArtifactCollectionMembershipFileUrlsProjectArtifactCollection,
+    ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership,
+)
 from .artifact_collection_membership_files import (
     ArtifactCollectionMembershipFiles,
     ArtifactCollectionMembershipFilesProject,
     ArtifactCollectionMembershipFilesProjectArtifactCollection,
     ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembership,
 )
+from .artifact_created_by import (
+    ArtifactCreatedBy,
+    ArtifactCreatedByArtifact,
+    ArtifactCreatedByArtifactCreatedByRun,
+    ArtifactCreatedByArtifactCreatedByRunProject,
+    ArtifactCreatedByArtifactCreatedByUser,
+)
+from .artifact_file_urls import ArtifactFileUrls, ArtifactFileUrlsArtifact
+from .artifact_type import (
+    ArtifactType,
+    ArtifactTypeProject,
+    ArtifactTypeProjectArtifact,
+    ArtifactTypeProjectArtifactArtifactType,
+)
+from .artifact_used_by import (
+    ArtifactUsedBy,
+    ArtifactUsedByArtifact,
+    ArtifactUsedByArtifactUsedBy,
+    ArtifactUsedByArtifactUsedByEdges,
+    ArtifactUsedByArtifactUsedByEdgesNode,
+    ArtifactUsedByArtifactUsedByEdgesNodeProject,
+)
 from .artifact_version_files import (
     ArtifactVersionFiles,
     ArtifactVersionFilesProject,
     ArtifactVersionFilesProjectArtifactType,
     ArtifactVersionFilesProjectArtifactTypeArtifact,
 )
+from .artifact_via_membership_by_name import (
+    ArtifactViaMembershipByName,
+    ArtifactViaMembershipByNameProject,
+)
 from .create_artifact_collection_tag_assignments import (
     CreateArtifactCollectionTagAssignments,
     CreateArtifactCollectionTagAssignmentsCreateArtifactCollectionTagAssignments,
     CreateArtifactCollectionTagAssignmentsCreateArtifactCollectionTagAssignmentsTags,
 )
 from .delete_aliases import DeleteAliases, DeleteAliasesDeleteAliases
+from .delete_artifact import (
+    DeleteArtifact,
+    DeleteArtifactDeleteArtifact,
+    DeleteArtifactDeleteArtifactArtifact,
+)
 from .delete_artifact_collection_tag_assignments import (
     DeleteArtifactCollectionTagAssignments,
     DeleteArtifactCollectionTagAssignmentsDeleteArtifactCollectionTagAssignments,
@@ -33,7 +72,14 @@ from .delete_artifact_sequence import (
     DeleteArtifactSequenceDeleteArtifactSequence,
     DeleteArtifactSequenceDeleteArtifactSequenceArtifactCollection,
 )
-from .enums import ArtifactCollectionState, ArtifactState
+from .enums import ArtifactCollectionState, ArtifactCollectionType, ArtifactState
+from .fetch_artifact_manifest import (
+    FetchArtifactManifest,
+    FetchArtifactManifestProject,
+    FetchArtifactManifestProjectArtifact,
+    FetchArtifactManifestProjectArtifactCurrentManifest,
+    FetchArtifactManifestProjectArtifactCurrentManifestFile,
+)
 from .fetch_linked_artifacts import (
     FetchLinkedArtifacts,
     FetchLinkedArtifactsArtifact,
@@ -44,6 +90,11 @@ from .fetch_linked_artifacts import (
     FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection,
     FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollectionProject,
 )
+from .fetch_registries import (
+    FetchRegistries,
+    FetchRegistriesOrganization,
+    FetchRegistriesOrganizationOrgEntity,
+)
 from .fragments import (
     ArtifactCollectionsFragment,
     ArtifactCollectionsFragmentEdges,
@@ -53,12 +104,13 @@ from .fragments import (
     ArtifactFragmentAliases,
     ArtifactFragmentAliasesArtifactCollection,
     ArtifactFragmentAliasesArtifactCollectionProject,
-    ArtifactFragmentArtifactSequence,
-    ArtifactFragmentArtifactSequenceProject,
-    ArtifactFragmentArtifactType,
-    ArtifactFragmentCurrentManifest,
-    ArtifactFragmentCurrentManifestFile,
-    ArtifactFragmentTags,
+    ArtifactFragmentWithoutAliases,
+    ArtifactFragmentWithoutAliasesArtifactSequence,
+    ArtifactFragmentWithoutAliasesArtifactSequenceProject,
+    ArtifactFragmentWithoutAliasesArtifactType,
+    ArtifactFragmentWithoutAliasesCurrentManifest,
+    ArtifactFragmentWithoutAliasesCurrentManifestFile,
+    ArtifactFragmentWithoutAliasesTags,
     ArtifactPortfolioTypeFields,
     ArtifactSequenceTypeFields,
     ArtifactsFragment,
@@ -72,6 +124,41 @@ from .fragments import (
     FilesFragmentEdges,
     FilesFragmentEdgesNode,
     FilesFragmentPageInfo,
+    FileUrlsFragment,
+    FileUrlsFragmentEdges,
+    FileUrlsFragmentEdgesNode,
+    FileUrlsFragmentPageInfo,
+    MembershipWithArtifact,
+    MembershipWithArtifactArtifactCollection,
+    MembershipWithArtifactArtifactCollectionProject,
+    RegistriesPage,
+    RegistriesPageEdges,
+    RegistriesPagePageInfo,
+    RegistryCollectionsPage,
+    RegistryCollectionsPageEdges,
+    RegistryCollectionsPageEdgesNode,
+    RegistryCollectionsPageEdgesNodeAliases,
+    RegistryCollectionsPageEdgesNodeAliasesEdges,
+    RegistryCollectionsPageEdgesNodeAliasesEdgesNode,
+    RegistryCollectionsPageEdgesNodeDefaultArtifactType,
+    RegistryCollectionsPageEdgesNodeProject,
+    RegistryCollectionsPageEdgesNodeProjectEntity,
+    RegistryCollectionsPageEdgesNodeTags,
+    RegistryCollectionsPageEdgesNodeTagsEdges,
+    RegistryCollectionsPageEdgesNodeTagsEdgesNode,
+    RegistryCollectionsPagePageInfo,
+    RegistryFragment,
+    RegistryFragmentArtifactTypes,
+    RegistryFragmentArtifactTypesEdges,
+    RegistryFragmentArtifactTypesEdgesNode,
+    RegistryVersionsPage,
+    RegistryVersionsPageEdges,
+    RegistryVersionsPageEdgesNode,
+    RegistryVersionsPageEdgesNodeAliases,
+    RegistryVersionsPageEdgesNodeArtifactCollection,
+    RegistryVersionsPageEdgesNodeArtifactCollectionProject,
+    RegistryVersionsPageEdgesNodeArtifactCollectionProjectEntity,
+    RegistryVersionsPagePageInfo,
 )
 from .input_types import (
     ArtifactAliasInput,
@@ -87,14 +174,25 @@ from .move_artifact_collection import (
 )
 from .operations import (
     ADD_ALIASES_GQL,
+    ARTIFACT_BY_ID_GQL,
+    ARTIFACT_BY_NAME_GQL,
+    ARTIFACT_COLLECTION_MEMBERSHIP_FILE_URLS_GQL,
     ARTIFACT_COLLECTION_MEMBERSHIP_FILES_GQL,
+    ARTIFACT_CREATED_BY_GQL,
+    ARTIFACT_FILE_URLS_GQL,
+    ARTIFACT_TYPE_GQL,
+    ARTIFACT_USED_BY_GQL,
     ARTIFACT_VERSION_FILES_GQL,
+    ARTIFACT_VIA_MEMBERSHIP_BY_NAME_GQL,
     CREATE_ARTIFACT_COLLECTION_TAG_ASSIGNMENTS_GQL,
     DELETE_ALIASES_GQL,
     DELETE_ARTIFACT_COLLECTION_TAG_ASSIGNMENTS_GQL,
+    DELETE_ARTIFACT_GQL,
     DELETE_ARTIFACT_PORTFOLIO_GQL,
     DELETE_ARTIFACT_SEQUENCE_GQL,
+    FETCH_ARTIFACT_MANIFEST_GQL,
     FETCH_LINKED_ARTIFACTS_GQL,
+    FETCH_REGISTRIES_GQL,
     LINK_ARTIFACT_GQL,
     MOVE_ARTIFACT_COLLECTION_GQL,
     PROJECT_ARTIFACT_COLLECTION_GQL,
@@ -102,8 +200,11 @@ from .operations import (
     PROJECT_ARTIFACT_TYPE_GQL,
     PROJECT_ARTIFACT_TYPES_GQL,
     PROJECT_ARTIFACTS_GQL,
+    REGISTRY_COLLECTIONS_GQL,
+    REGISTRY_VERSIONS_GQL,
     RUN_INPUT_ARTIFACTS_GQL,
     RUN_OUTPUT_ARTIFACTS_GQL,
+    UNLINK_ARTIFACT_GQL,
     UPDATE_ARTIFACT_GQL,
     UPDATE_ARTIFACT_PORTFOLIO_GQL,
     UPDATE_ARTIFACT_SEQUENCE_GQL,
@@ -135,6 +236,16 @@ from .project_artifacts import (
     ProjectArtifactsProjectArtifactType,
     ProjectArtifactsProjectArtifactTypeArtifactCollection,
 )
+from .registry_collections import (
+    RegistryCollections,
+    RegistryCollectionsOrganization,
+    RegistryCollectionsOrganizationOrgEntity,
+)
+from .registry_versions import (
+    RegistryVersions,
+    RegistryVersionsOrganization,
+    RegistryVersionsOrganizationOrgEntity,
+)
 from .run_input_artifacts import (
     RunInputArtifacts,
     RunInputArtifactsProject,
@@ -151,6 +262,7 @@ from .run_output_artifacts import (
     RunOutputArtifactsProjectRunOutputArtifactsEdges,
     RunOutputArtifactsProjectRunOutputArtifactsPageInfo,
 )
+from .unlink_artifact import UnlinkArtifact, UnlinkArtifactUnlinkArtifact
 from .update_artifact import UpdateArtifact, UpdateArtifactUpdateArtifact
 from .update_artifact_portfolio import (
     UpdateArtifactPortfolio,
@@ -165,14 +277,25 @@ from .update_artifact_sequence import (
 
 __all__ = [
     "ADD_ALIASES_GQL",
+    "ARTIFACT_BY_ID_GQL",
+    "ARTIFACT_BY_NAME_GQL",
     "ARTIFACT_COLLECTION_MEMBERSHIP_FILES_GQL",
+    "ARTIFACT_COLLECTION_MEMBERSHIP_FILE_URLS_GQL",
+    "ARTIFACT_CREATED_BY_GQL",
+    "ARTIFACT_FILE_URLS_GQL",
+    "ARTIFACT_TYPE_GQL",
+    "ARTIFACT_USED_BY_GQL",
     "ARTIFACT_VERSION_FILES_GQL",
+    "ARTIFACT_VIA_MEMBERSHIP_BY_NAME_GQL",
     "CREATE_ARTIFACT_COLLECTION_TAG_ASSIGNMENTS_GQL",
     "DELETE_ALIASES_GQL",
     "DELETE_ARTIFACT_COLLECTION_TAG_ASSIGNMENTS_GQL",
+    "DELETE_ARTIFACT_GQL",
     "DELETE_ARTIFACT_PORTFOLIO_GQL",
     "DELETE_ARTIFACT_SEQUENCE_GQL",
+    "FETCH_ARTIFACT_MANIFEST_GQL",
     "FETCH_LINKED_ARTIFACTS_GQL",
+    "FETCH_REGISTRIES_GQL",
     "LINK_ARTIFACT_GQL",
     "MOVE_ARTIFACT_COLLECTION_GQL",
     "PROJECT_ARTIFACTS_GQL",
@@ -180,8 +303,11 @@ __all__ = [
     "PROJECT_ARTIFACT_COLLECTION_GQL",
     "PROJECT_ARTIFACT_TYPES_GQL",
     "PROJECT_ARTIFACT_TYPE_GQL",
+    "REGISTRY_COLLECTIONS_GQL",
+    "REGISTRY_VERSIONS_GQL",
     "RUN_INPUT_ARTIFACTS_GQL",
     "RUN_OUTPUT_ARTIFACTS_GQL",
+    "UNLINK_ARTIFACT_GQL",
     "UPDATE_ARTIFACT_GQL",
     "UPDATE_ARTIFACT_PORTFOLIO_GQL",
     "UPDATE_ARTIFACT_SEQUENCE_GQL",
@@ -228,6 +354,12 @@ __all__ = [
     "ArtifactCollectionMembershipFilesProject",
     "ArtifactCollectionMembershipFilesProjectArtifactCollection",
     "ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembership",
+    "ArtifactCollectionMembershipFileUrls",
+    "ArtifactCollectionMembershipFileUrlsProject",
+    "ArtifactCollectionMembershipFileUrlsProjectArtifactCollection",
+    "ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership",
+    "ArtifactFileUrls",
+    "ArtifactFileUrlsArtifact",
     "ProjectArtifactTypes",
     "ProjectArtifactTypesProject",
     "ProjectArtifactType",
@@ -256,14 +388,53 @@ __all__ = [
     "FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeAliases",
     "FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection",
     "FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollectionProject",
+    "FetchArtifactManifest",
+    "FetchArtifactManifestProject",
+    "FetchArtifactManifestProjectArtifact",
+    "FetchArtifactManifestProjectArtifactCurrentManifest",
+    "FetchArtifactManifestProjectArtifactCurrentManifestFile",
+    "ArtifactByID",
+    "ArtifactByName",
+    "ArtifactByNameProject",
+    "ArtifactViaMembershipByName",
+    "ArtifactViaMembershipByNameProject",
+    "ArtifactUsedBy",
+    "ArtifactUsedByArtifact",
+    "ArtifactUsedByArtifactUsedBy",
+    "ArtifactUsedByArtifactUsedByEdges",
+    "ArtifactUsedByArtifactUsedByEdgesNode",
+    "ArtifactUsedByArtifactUsedByEdgesNodeProject",
+    "ArtifactCreatedBy",
+    "ArtifactCreatedByArtifact",
+    "ArtifactCreatedByArtifactCreatedByRun",
+    "ArtifactCreatedByArtifactCreatedByRunProject",
+    "ArtifactCreatedByArtifactCreatedByUser",
+    "ArtifactType",
+    "ArtifactTypeProject",
+    "ArtifactTypeProjectArtifact",
+    "ArtifactTypeProjectArtifactArtifactType",
     "AddAliases",
     "AddAliasesAddAliases",
     "DeleteAliases",
     "DeleteAliasesDeleteAliases",
     "UpdateArtifact",
     "UpdateArtifactUpdateArtifact",
+    "DeleteArtifact",
+    "DeleteArtifactDeleteArtifact",
+    "DeleteArtifactDeleteArtifactArtifact",
     "LinkArtifact",
     "LinkArtifactLinkArtifact",
+    "UnlinkArtifact",
+    "UnlinkArtifactUnlinkArtifact",
+    "RegistryVersions",
+    "RegistryVersionsOrganization",
+    "RegistryVersionsOrganizationOrgEntity",
+    "RegistryCollections",
+    "RegistryCollectionsOrganization",
+    "RegistryCollectionsOrganizationOrgEntity",
+    "FetchRegistries",
+    "FetchRegistriesOrganization",
+    "FetchRegistriesOrganizationOrgEntity",
     "ArtifactAliasInput",
     "ArtifactCollectionAliasInput",
     "LinkArtifactInput",
@@ -276,12 +447,13 @@ __all__ = [
     "ArtifactFragmentAliases",
     "ArtifactFragmentAliasesArtifactCollection",
     "ArtifactFragmentAliasesArtifactCollectionProject",
-    "ArtifactFragmentArtifactSequence",
-    "ArtifactFragmentArtifactSequenceProject",
-    "ArtifactFragmentArtifactType",
-    "ArtifactFragmentCurrentManifest",
-    "ArtifactFragmentCurrentManifestFile",
-    "ArtifactFragmentTags",
+    "ArtifactFragmentWithoutAliases",
+    "ArtifactFragmentWithoutAliasesArtifactSequence",
+    "ArtifactFragmentWithoutAliasesArtifactSequenceProject",
+    "ArtifactFragmentWithoutAliasesArtifactType",
+    "ArtifactFragmentWithoutAliasesCurrentManifest",
+    "ArtifactFragmentWithoutAliasesCurrentManifestFile",
+    "ArtifactFragmentWithoutAliasesTags",
     "ArtifactPortfolioTypeFields",
     "ArtifactSequenceTypeFields",
     "ArtifactTypeFragment",
@@ -291,10 +463,46 @@ __all__ = [
     "ArtifactsFragment",
     "ArtifactsFragmentEdges",
     "ArtifactsFragmentPageInfo",
+    "FileUrlsFragment",
+    "FileUrlsFragmentEdges",
+    "FileUrlsFragmentEdgesNode",
+    "FileUrlsFragmentPageInfo",
     "FilesFragment",
     "FilesFragmentEdges",
     "FilesFragmentEdgesNode",
     "FilesFragmentPageInfo",
+    "MembershipWithArtifact",
+    "MembershipWithArtifactArtifactCollection",
+    "MembershipWithArtifactArtifactCollectionProject",
+    "RegistriesPage",
+    "RegistriesPageEdges",
+    "RegistriesPagePageInfo",
+    "RegistryCollectionsPage",
+    "RegistryCollectionsPageEdges",
+    "RegistryCollectionsPageEdgesNode",
+    "RegistryCollectionsPageEdgesNodeAliases",
+    "RegistryCollectionsPageEdgesNodeAliasesEdges",
+    "RegistryCollectionsPageEdgesNodeAliasesEdgesNode",
+    "RegistryCollectionsPageEdgesNodeDefaultArtifactType",
+    "RegistryCollectionsPageEdgesNodeProject",
+    "RegistryCollectionsPageEdgesNodeProjectEntity",
+    "RegistryCollectionsPageEdgesNodeTags",
+    "RegistryCollectionsPageEdgesNodeTagsEdges",
+    "RegistryCollectionsPageEdgesNodeTagsEdgesNode",
+    "RegistryCollectionsPagePageInfo",
+    "RegistryFragment",
+    "RegistryFragmentArtifactTypes",
+    "RegistryFragmentArtifactTypesEdges",
+    "RegistryFragmentArtifactTypesEdgesNode",
+    "RegistryVersionsPage",
+    "RegistryVersionsPageEdges",
+    "RegistryVersionsPageEdgesNode",
+    "RegistryVersionsPageEdgesNodeAliases",
+    "RegistryVersionsPageEdgesNodeArtifactCollection",
+    "RegistryVersionsPageEdgesNodeArtifactCollectionProject",
+    "RegistryVersionsPageEdgesNodeArtifactCollectionProjectEntity",
+    "RegistryVersionsPagePageInfo",
     "ArtifactCollectionState",
+    "ArtifactCollectionType",
     "ArtifactState",
 ]

wandb/sdk/artifacts/_generated/artifact_by_id.py

@@ -0,0 +1,17 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactFragment
+
+
+class ArtifactByID(GQLBase):
+    artifact: Optional[ArtifactFragment]
+
+
+ArtifactByID.model_rebuild()

wandb/sdk/artifacts/_generated/artifact_by_name.py

@@ -0,0 +1,22 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactFragment
+
+
+class ArtifactByName(GQLBase):
+    project: Optional[ArtifactByNameProject]
+
+
+class ArtifactByNameProject(GQLBase):
+    artifact: Optional[ArtifactFragment]
+
+
+ArtifactByName.model_rebuild()
+ArtifactByNameProject.model_rebuild()

wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py

@@ -0,0 +1,43 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Literal, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, Typename
+
+from .fragments import FileUrlsFragment
+
+
+class ArtifactCollectionMembershipFileUrls(GQLBase):
+    project: Optional[ArtifactCollectionMembershipFileUrlsProject]
+
+
+class ArtifactCollectionMembershipFileUrlsProject(GQLBase):
+    artifact_collection: Optional[
+        ArtifactCollectionMembershipFileUrlsProjectArtifactCollection
+    ] = Field(alias="artifactCollection")
+
+
+class ArtifactCollectionMembershipFileUrlsProjectArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    artifact_membership: Optional[
+        ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership
+    ] = Field(alias="artifactMembership")
+
+
+class ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership(
+    GQLBase
+):
+    files: Optional[FileUrlsFragment]
+
+
+ArtifactCollectionMembershipFileUrls.model_rebuild()
+ArtifactCollectionMembershipFileUrlsProject.model_rebuild()
+ArtifactCollectionMembershipFileUrlsProjectArtifactCollection.model_rebuild()
+ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership.model_rebuild()

wandb/sdk/artifacts/_generated/artifact_created_by.py

@@ -0,0 +1,47 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Literal, Optional, Union
+
+from pydantic import Field
+from typing_extensions import Annotated
+
+from wandb._pydantic import GQLBase, Typename
+
+
+class ArtifactCreatedBy(GQLBase):
+    artifact: Optional[ArtifactCreatedByArtifact]
+
+
+class ArtifactCreatedByArtifact(GQLBase):
+    created_by: Optional[
+        Annotated[
+            Union[
+                ArtifactCreatedByArtifactCreatedByRun,
+                ArtifactCreatedByArtifactCreatedByUser,
+            ],
+            Field(discriminator="typename__"),
+        ]
+    ] = Field(alias="createdBy")
+
+
+class ArtifactCreatedByArtifactCreatedByRun(GQLBase):
+    typename__: Typename[Literal["Run"]]
+    name: str
+    project: Optional[ArtifactCreatedByArtifactCreatedByRunProject]
+
+
+class ArtifactCreatedByArtifactCreatedByRunProject(GQLBase):
+    name: str
+    entity_name: str = Field(alias="entityName")
+
+
+class ArtifactCreatedByArtifactCreatedByUser(GQLBase):
+    typename__: Typename[Literal["User"]]
+
+
+ArtifactCreatedBy.model_rebuild()
+ArtifactCreatedByArtifact.model_rebuild()
+ArtifactCreatedByArtifactCreatedByRun.model_rebuild()

wandb/sdk/artifacts/_generated/artifact_file_urls.py

@@ -0,0 +1,22 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from wandb._pydantic import GQLBase
+
+from .fragments import FileUrlsFragment
+
+
+class ArtifactFileUrls(GQLBase):
+    artifact: Optional[ArtifactFileUrlsArtifact]
+
+
+class ArtifactFileUrlsArtifact(GQLBase):
+    files: Optional[FileUrlsFragment]
+
+
+ArtifactFileUrls.model_rebuild()
+ArtifactFileUrlsArtifact.model_rebuild()

wandb/sdk/artifacts/_generated/artifact_type.py

@@ -0,0 +1,31 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+
+class ArtifactType(GQLBase):
+    project: Optional[ArtifactTypeProject]
+
+
+class ArtifactTypeProject(GQLBase):
+    artifact: Optional[ArtifactTypeProjectArtifact]
+
+
+class ArtifactTypeProjectArtifact(GQLBase):
+    artifact_type: ArtifactTypeProjectArtifactArtifactType = Field(alias="artifactType")
+
+
+class ArtifactTypeProjectArtifactArtifactType(GQLBase):
+    name: str
+
+
+ArtifactType.model_rebuild()
+ArtifactTypeProject.model_rebuild()
+ArtifactTypeProjectArtifact.model_rebuild()