wandb 0.18.2__py3-none-musllinux_1_2_x86_64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (827) hide show
  1. package_readme.md +89 -0
  2. wandb/__init__.py +245 -0
  3. wandb/__init__.pyi +1139 -0
  4. wandb/__main__.py +3 -0
  5. wandb/_globals.py +19 -0
  6. wandb/agents/__init__.py +0 -0
  7. wandb/agents/pyagent.py +363 -0
  8. wandb/analytics/__init__.py +3 -0
  9. wandb/analytics/sentry.py +266 -0
  10. wandb/apis/__init__.py +48 -0
  11. wandb/apis/attrs.py +40 -0
  12. wandb/apis/importers/__init__.py +1 -0
  13. wandb/apis/importers/internals/internal.py +385 -0
  14. wandb/apis/importers/internals/protocols.py +99 -0
  15. wandb/apis/importers/internals/util.py +78 -0
  16. wandb/apis/importers/mlflow.py +254 -0
  17. wandb/apis/importers/validation.py +108 -0
  18. wandb/apis/importers/wandb.py +1603 -0
  19. wandb/apis/internal.py +232 -0
  20. wandb/apis/normalize.py +89 -0
  21. wandb/apis/paginator.py +81 -0
  22. wandb/apis/public/__init__.py +34 -0
  23. wandb/apis/public/api.py +1305 -0
  24. wandb/apis/public/artifacts.py +1090 -0
  25. wandb/apis/public/const.py +4 -0
  26. wandb/apis/public/files.py +195 -0
  27. wandb/apis/public/history.py +149 -0
  28. wandb/apis/public/jobs.py +659 -0
  29. wandb/apis/public/projects.py +154 -0
  30. wandb/apis/public/query_generator.py +166 -0
  31. wandb/apis/public/reports.py +469 -0
  32. wandb/apis/public/runs.py +914 -0
  33. wandb/apis/public/sweeps.py +240 -0
  34. wandb/apis/public/teams.py +198 -0
  35. wandb/apis/public/users.py +136 -0
  36. wandb/apis/reports/__init__.py +1 -0
  37. wandb/apis/reports/v1/__init__.py +8 -0
  38. wandb/apis/reports/v2/__init__.py +8 -0
  39. wandb/apis/workspaces/__init__.py +8 -0
  40. wandb/beta/workflows.py +288 -0
  41. wandb/bin/nvidia_gpu_stats +0 -0
  42. wandb/bin/wandb-core +0 -0
  43. wandb/cli/__init__.py +0 -0
  44. wandb/cli/cli.py +3004 -0
  45. wandb/data_types.py +63 -0
  46. wandb/docker/__init__.py +342 -0
  47. wandb/docker/auth.py +436 -0
  48. wandb/docker/wandb-entrypoint.sh +33 -0
  49. wandb/docker/www_authenticate.py +94 -0
  50. wandb/env.py +514 -0
  51. wandb/errors/__init__.py +17 -0
  52. wandb/errors/errors.py +37 -0
  53. wandb/errors/term.py +103 -0
  54. wandb/errors/util.py +57 -0
  55. wandb/errors/warnings.py +2 -0
  56. wandb/filesync/__init__.py +0 -0
  57. wandb/filesync/dir_watcher.py +403 -0
  58. wandb/filesync/stats.py +100 -0
  59. wandb/filesync/step_checksum.py +142 -0
  60. wandb/filesync/step_prepare.py +179 -0
  61. wandb/filesync/step_upload.py +290 -0
  62. wandb/filesync/upload_job.py +142 -0
  63. wandb/integration/__init__.py +0 -0
  64. wandb/integration/catboost/__init__.py +5 -0
  65. wandb/integration/catboost/catboost.py +178 -0
  66. wandb/integration/cohere/__init__.py +3 -0
  67. wandb/integration/cohere/cohere.py +21 -0
  68. wandb/integration/cohere/resolver.py +347 -0
  69. wandb/integration/diffusers/__init__.py +3 -0
  70. wandb/integration/diffusers/autologger.py +76 -0
  71. wandb/integration/diffusers/pipeline_resolver.py +50 -0
  72. wandb/integration/diffusers/resolvers/__init__.py +9 -0
  73. wandb/integration/diffusers/resolvers/multimodal.py +882 -0
  74. wandb/integration/diffusers/resolvers/utils.py +102 -0
  75. wandb/integration/fastai/__init__.py +249 -0
  76. wandb/integration/gym/__init__.py +105 -0
  77. wandb/integration/huggingface/__init__.py +3 -0
  78. wandb/integration/huggingface/huggingface.py +18 -0
  79. wandb/integration/huggingface/resolver.py +213 -0
  80. wandb/integration/keras/__init__.py +11 -0
  81. wandb/integration/keras/callbacks/__init__.py +5 -0
  82. wandb/integration/keras/callbacks/metrics_logger.py +136 -0
  83. wandb/integration/keras/callbacks/model_checkpoint.py +195 -0
  84. wandb/integration/keras/callbacks/tables_builder.py +226 -0
  85. wandb/integration/keras/keras.py +1091 -0
  86. wandb/integration/kfp/__init__.py +6 -0
  87. wandb/integration/kfp/helpers.py +28 -0
  88. wandb/integration/kfp/kfp_patch.py +324 -0
  89. wandb/integration/kfp/wandb_logging.py +182 -0
  90. wandb/integration/langchain/__init__.py +3 -0
  91. wandb/integration/langchain/wandb_tracer.py +48 -0
  92. wandb/integration/lightgbm/__init__.py +239 -0
  93. wandb/integration/lightning/__init__.py +0 -0
  94. wandb/integration/lightning/fabric/__init__.py +3 -0
  95. wandb/integration/lightning/fabric/logger.py +762 -0
  96. wandb/integration/magic.py +556 -0
  97. wandb/integration/metaflow/__init__.py +3 -0
  98. wandb/integration/metaflow/metaflow.py +383 -0
  99. wandb/integration/openai/__init__.py +3 -0
  100. wandb/integration/openai/fine_tuning.py +480 -0
  101. wandb/integration/openai/openai.py +22 -0
  102. wandb/integration/openai/resolver.py +240 -0
  103. wandb/integration/prodigy/__init__.py +3 -0
  104. wandb/integration/prodigy/prodigy.py +299 -0
  105. wandb/integration/sacred/__init__.py +117 -0
  106. wandb/integration/sagemaker/__init__.py +12 -0
  107. wandb/integration/sagemaker/auth.py +28 -0
  108. wandb/integration/sagemaker/config.py +49 -0
  109. wandb/integration/sagemaker/files.py +3 -0
  110. wandb/integration/sagemaker/resources.py +34 -0
  111. wandb/integration/sb3/__init__.py +3 -0
  112. wandb/integration/sb3/sb3.py +153 -0
  113. wandb/integration/sklearn/__init__.py +37 -0
  114. wandb/integration/sklearn/calculate/__init__.py +32 -0
  115. wandb/integration/sklearn/calculate/calibration_curves.py +125 -0
  116. wandb/integration/sklearn/calculate/class_proportions.py +68 -0
  117. wandb/integration/sklearn/calculate/confusion_matrix.py +93 -0
  118. wandb/integration/sklearn/calculate/decision_boundaries.py +40 -0
  119. wandb/integration/sklearn/calculate/elbow_curve.py +55 -0
  120. wandb/integration/sklearn/calculate/feature_importances.py +67 -0
  121. wandb/integration/sklearn/calculate/learning_curve.py +64 -0
  122. wandb/integration/sklearn/calculate/outlier_candidates.py +69 -0
  123. wandb/integration/sklearn/calculate/residuals.py +86 -0
  124. wandb/integration/sklearn/calculate/silhouette.py +118 -0
  125. wandb/integration/sklearn/calculate/summary_metrics.py +62 -0
  126. wandb/integration/sklearn/plot/__init__.py +35 -0
  127. wandb/integration/sklearn/plot/classifier.py +329 -0
  128. wandb/integration/sklearn/plot/clusterer.py +146 -0
  129. wandb/integration/sklearn/plot/regressor.py +121 -0
  130. wandb/integration/sklearn/plot/shared.py +91 -0
  131. wandb/integration/sklearn/utils.py +183 -0
  132. wandb/integration/tensorboard/__init__.py +10 -0
  133. wandb/integration/tensorboard/log.py +355 -0
  134. wandb/integration/tensorboard/monkeypatch.py +185 -0
  135. wandb/integration/tensorflow/__init__.py +5 -0
  136. wandb/integration/tensorflow/estimator_hook.py +54 -0
  137. wandb/integration/torch/__init__.py +0 -0
  138. wandb/integration/torch/wandb_torch.py +554 -0
  139. wandb/integration/ultralytics/__init__.py +11 -0
  140. wandb/integration/ultralytics/bbox_utils.py +208 -0
  141. wandb/integration/ultralytics/callback.py +524 -0
  142. wandb/integration/ultralytics/classification_utils.py +83 -0
  143. wandb/integration/ultralytics/mask_utils.py +202 -0
  144. wandb/integration/ultralytics/pose_utils.py +103 -0
  145. wandb/integration/xgboost/__init__.py +11 -0
  146. wandb/integration/xgboost/xgboost.py +189 -0
  147. wandb/integration/yolov8/__init__.py +0 -0
  148. wandb/integration/yolov8/yolov8.py +284 -0
  149. wandb/jupyter.py +515 -0
  150. wandb/magic.py +3 -0
  151. wandb/mpmain/__init__.py +0 -0
  152. wandb/mpmain/__main__.py +1 -0
  153. wandb/old/__init__.py +0 -0
  154. wandb/old/core.py +53 -0
  155. wandb/old/settings.py +173 -0
  156. wandb/old/summary.py +440 -0
  157. wandb/plot/__init__.py +19 -0
  158. wandb/plot/bar.py +45 -0
  159. wandb/plot/confusion_matrix.py +100 -0
  160. wandb/plot/histogram.py +39 -0
  161. wandb/plot/line.py +43 -0
  162. wandb/plot/line_series.py +88 -0
  163. wandb/plot/pr_curve.py +136 -0
  164. wandb/plot/roc_curve.py +118 -0
  165. wandb/plot/scatter.py +32 -0
  166. wandb/plot/utils.py +183 -0
  167. wandb/plot/viz.py +123 -0
  168. wandb/proto/__init__.py +0 -0
  169. wandb/proto/v3/__init__.py +0 -0
  170. wandb/proto/v3/wandb_base_pb2.py +55 -0
  171. wandb/proto/v3/wandb_internal_pb2.py +1608 -0
  172. wandb/proto/v3/wandb_server_pb2.py +208 -0
  173. wandb/proto/v3/wandb_settings_pb2.py +112 -0
  174. wandb/proto/v3/wandb_telemetry_pb2.py +106 -0
  175. wandb/proto/v4/__init__.py +0 -0
  176. wandb/proto/v4/wandb_base_pb2.py +30 -0
  177. wandb/proto/v4/wandb_internal_pb2.py +360 -0
  178. wandb/proto/v4/wandb_server_pb2.py +63 -0
  179. wandb/proto/v4/wandb_settings_pb2.py +45 -0
  180. wandb/proto/v4/wandb_telemetry_pb2.py +41 -0
  181. wandb/proto/v5/wandb_base_pb2.py +31 -0
  182. wandb/proto/v5/wandb_internal_pb2.py +361 -0
  183. wandb/proto/v5/wandb_server_pb2.py +64 -0
  184. wandb/proto/v5/wandb_settings_pb2.py +46 -0
  185. wandb/proto/v5/wandb_telemetry_pb2.py +42 -0
  186. wandb/proto/wandb_base_pb2.py +10 -0
  187. wandb/proto/wandb_deprecated.py +53 -0
  188. wandb/proto/wandb_generate_deprecated.py +34 -0
  189. wandb/proto/wandb_generate_proto.py +49 -0
  190. wandb/proto/wandb_internal_pb2.py +16 -0
  191. wandb/proto/wandb_server_pb2.py +10 -0
  192. wandb/proto/wandb_settings_pb2.py +10 -0
  193. wandb/proto/wandb_telemetry_pb2.py +10 -0
  194. wandb/py.typed +0 -0
  195. wandb/sdk/__init__.py +37 -0
  196. wandb/sdk/artifacts/__init__.py +0 -0
  197. wandb/sdk/artifacts/_validators.py +90 -0
  198. wandb/sdk/artifacts/artifact.py +2389 -0
  199. wandb/sdk/artifacts/artifact_download_logger.py +43 -0
  200. wandb/sdk/artifacts/artifact_file_cache.py +253 -0
  201. wandb/sdk/artifacts/artifact_instance_cache.py +17 -0
  202. wandb/sdk/artifacts/artifact_manifest.py +74 -0
  203. wandb/sdk/artifacts/artifact_manifest_entry.py +249 -0
  204. wandb/sdk/artifacts/artifact_manifests/__init__.py +0 -0
  205. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +92 -0
  206. wandb/sdk/artifacts/artifact_saver.py +269 -0
  207. wandb/sdk/artifacts/artifact_state.py +11 -0
  208. wandb/sdk/artifacts/artifact_ttl.py +7 -0
  209. wandb/sdk/artifacts/exceptions.py +57 -0
  210. wandb/sdk/artifacts/staging.py +25 -0
  211. wandb/sdk/artifacts/storage_handler.py +62 -0
  212. wandb/sdk/artifacts/storage_handlers/__init__.py +0 -0
  213. wandb/sdk/artifacts/storage_handlers/azure_handler.py +208 -0
  214. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +228 -0
  215. wandb/sdk/artifacts/storage_handlers/http_handler.py +114 -0
  216. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +141 -0
  217. wandb/sdk/artifacts/storage_handlers/multi_handler.py +56 -0
  218. wandb/sdk/artifacts/storage_handlers/s3_handler.py +300 -0
  219. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +72 -0
  220. wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +135 -0
  221. wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +74 -0
  222. wandb/sdk/artifacts/storage_layout.py +6 -0
  223. wandb/sdk/artifacts/storage_policies/__init__.py +4 -0
  224. wandb/sdk/artifacts/storage_policies/register.py +1 -0
  225. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +378 -0
  226. wandb/sdk/artifacts/storage_policy.py +72 -0
  227. wandb/sdk/backend/__init__.py +0 -0
  228. wandb/sdk/backend/backend.py +222 -0
  229. wandb/sdk/data_types/__init__.py +0 -0
  230. wandb/sdk/data_types/_dtypes.py +914 -0
  231. wandb/sdk/data_types/_private.py +10 -0
  232. wandb/sdk/data_types/audio.py +165 -0
  233. wandb/sdk/data_types/base_types/__init__.py +0 -0
  234. wandb/sdk/data_types/base_types/json_metadata.py +55 -0
  235. wandb/sdk/data_types/base_types/media.py +315 -0
  236. wandb/sdk/data_types/base_types/wb_value.py +272 -0
  237. wandb/sdk/data_types/bokeh.py +70 -0
  238. wandb/sdk/data_types/graph.py +405 -0
  239. wandb/sdk/data_types/helper_types/__init__.py +0 -0
  240. wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +295 -0
  241. wandb/sdk/data_types/helper_types/classes.py +159 -0
  242. wandb/sdk/data_types/helper_types/image_mask.py +235 -0
  243. wandb/sdk/data_types/histogram.py +96 -0
  244. wandb/sdk/data_types/html.py +115 -0
  245. wandb/sdk/data_types/image.py +845 -0
  246. wandb/sdk/data_types/molecule.py +241 -0
  247. wandb/sdk/data_types/object_3d.py +474 -0
  248. wandb/sdk/data_types/plotly.py +82 -0
  249. wandb/sdk/data_types/saved_model.py +446 -0
  250. wandb/sdk/data_types/table.py +1204 -0
  251. wandb/sdk/data_types/trace_tree.py +438 -0
  252. wandb/sdk/data_types/utils.py +229 -0
  253. wandb/sdk/data_types/video.py +247 -0
  254. wandb/sdk/integration_utils/__init__.py +0 -0
  255. wandb/sdk/integration_utils/auto_logging.py +239 -0
  256. wandb/sdk/integration_utils/data_logging.py +475 -0
  257. wandb/sdk/interface/__init__.py +0 -0
  258. wandb/sdk/interface/constants.py +4 -0
  259. wandb/sdk/interface/interface.py +972 -0
  260. wandb/sdk/interface/interface_queue.py +59 -0
  261. wandb/sdk/interface/interface_relay.py +53 -0
  262. wandb/sdk/interface/interface_shared.py +537 -0
  263. wandb/sdk/interface/interface_sock.py +61 -0
  264. wandb/sdk/interface/message_future.py +27 -0
  265. wandb/sdk/interface/message_future_poll.py +50 -0
  266. wandb/sdk/interface/router.py +118 -0
  267. wandb/sdk/interface/router_queue.py +44 -0
  268. wandb/sdk/interface/router_relay.py +39 -0
  269. wandb/sdk/interface/router_sock.py +36 -0
  270. wandb/sdk/interface/summary_record.py +67 -0
  271. wandb/sdk/internal/__init__.py +0 -0
  272. wandb/sdk/internal/context.py +89 -0
  273. wandb/sdk/internal/datastore.py +297 -0
  274. wandb/sdk/internal/file_pusher.py +181 -0
  275. wandb/sdk/internal/file_stream.py +695 -0
  276. wandb/sdk/internal/flow_control.py +263 -0
  277. wandb/sdk/internal/handler.py +901 -0
  278. wandb/sdk/internal/internal.py +417 -0
  279. wandb/sdk/internal/internal_api.py +4358 -0
  280. wandb/sdk/internal/internal_util.py +100 -0
  281. wandb/sdk/internal/job_builder.py +629 -0
  282. wandb/sdk/internal/profiler.py +78 -0
  283. wandb/sdk/internal/progress.py +83 -0
  284. wandb/sdk/internal/run.py +25 -0
  285. wandb/sdk/internal/sample.py +70 -0
  286. wandb/sdk/internal/sender.py +1686 -0
  287. wandb/sdk/internal/sender_config.py +197 -0
  288. wandb/sdk/internal/settings_static.py +90 -0
  289. wandb/sdk/internal/system/__init__.py +0 -0
  290. wandb/sdk/internal/system/assets/__init__.py +27 -0
  291. wandb/sdk/internal/system/assets/aggregators.py +37 -0
  292. wandb/sdk/internal/system/assets/asset_registry.py +20 -0
  293. wandb/sdk/internal/system/assets/cpu.py +163 -0
  294. wandb/sdk/internal/system/assets/disk.py +210 -0
  295. wandb/sdk/internal/system/assets/gpu.py +416 -0
  296. wandb/sdk/internal/system/assets/gpu_amd.py +239 -0
  297. wandb/sdk/internal/system/assets/gpu_apple.py +177 -0
  298. wandb/sdk/internal/system/assets/interfaces.py +207 -0
  299. wandb/sdk/internal/system/assets/ipu.py +177 -0
  300. wandb/sdk/internal/system/assets/memory.py +166 -0
  301. wandb/sdk/internal/system/assets/network.py +125 -0
  302. wandb/sdk/internal/system/assets/open_metrics.py +299 -0
  303. wandb/sdk/internal/system/assets/tpu.py +154 -0
  304. wandb/sdk/internal/system/assets/trainium.py +399 -0
  305. wandb/sdk/internal/system/env_probe_helpers.py +13 -0
  306. wandb/sdk/internal/system/system_info.py +249 -0
  307. wandb/sdk/internal/system/system_monitor.py +229 -0
  308. wandb/sdk/internal/tb_watcher.py +518 -0
  309. wandb/sdk/internal/thread_local_settings.py +18 -0
  310. wandb/sdk/internal/writer.py +206 -0
  311. wandb/sdk/launch/__init__.py +14 -0
  312. wandb/sdk/launch/_launch.py +330 -0
  313. wandb/sdk/launch/_launch_add.py +255 -0
  314. wandb/sdk/launch/_project_spec.py +566 -0
  315. wandb/sdk/launch/agent/__init__.py +5 -0
  316. wandb/sdk/launch/agent/agent.py +924 -0
  317. wandb/sdk/launch/agent/config.py +296 -0
  318. wandb/sdk/launch/agent/job_status_tracker.py +53 -0
  319. wandb/sdk/launch/agent/run_queue_item_file_saver.py +45 -0
  320. wandb/sdk/launch/builder/__init__.py +0 -0
  321. wandb/sdk/launch/builder/abstract.py +156 -0
  322. wandb/sdk/launch/builder/build.py +297 -0
  323. wandb/sdk/launch/builder/context_manager.py +235 -0
  324. wandb/sdk/launch/builder/docker_builder.py +177 -0
  325. wandb/sdk/launch/builder/kaniko_builder.py +595 -0
  326. wandb/sdk/launch/builder/noop.py +58 -0
  327. wandb/sdk/launch/builder/templates/_wandb_bootstrap.py +188 -0
  328. wandb/sdk/launch/builder/templates/dockerfile.py +92 -0
  329. wandb/sdk/launch/create_job.py +528 -0
  330. wandb/sdk/launch/environment/abstract.py +29 -0
  331. wandb/sdk/launch/environment/aws_environment.py +322 -0
  332. wandb/sdk/launch/environment/azure_environment.py +105 -0
  333. wandb/sdk/launch/environment/gcp_environment.py +335 -0
  334. wandb/sdk/launch/environment/local_environment.py +66 -0
  335. wandb/sdk/launch/errors.py +19 -0
  336. wandb/sdk/launch/git_reference.py +109 -0
  337. wandb/sdk/launch/inputs/files.py +148 -0
  338. wandb/sdk/launch/inputs/internal.py +315 -0
  339. wandb/sdk/launch/inputs/manage.py +113 -0
  340. wandb/sdk/launch/inputs/schema.py +39 -0
  341. wandb/sdk/launch/loader.py +249 -0
  342. wandb/sdk/launch/registry/abstract.py +48 -0
  343. wandb/sdk/launch/registry/anon.py +29 -0
  344. wandb/sdk/launch/registry/azure_container_registry.py +124 -0
  345. wandb/sdk/launch/registry/elastic_container_registry.py +192 -0
  346. wandb/sdk/launch/registry/google_artifact_registry.py +219 -0
  347. wandb/sdk/launch/registry/local_registry.py +67 -0
  348. wandb/sdk/launch/runner/__init__.py +0 -0
  349. wandb/sdk/launch/runner/abstract.py +195 -0
  350. wandb/sdk/launch/runner/kubernetes_monitor.py +474 -0
  351. wandb/sdk/launch/runner/kubernetes_runner.py +963 -0
  352. wandb/sdk/launch/runner/local_container.py +301 -0
  353. wandb/sdk/launch/runner/local_process.py +78 -0
  354. wandb/sdk/launch/runner/sagemaker_runner.py +426 -0
  355. wandb/sdk/launch/runner/vertex_runner.py +230 -0
  356. wandb/sdk/launch/sweeps/__init__.py +39 -0
  357. wandb/sdk/launch/sweeps/scheduler.py +742 -0
  358. wandb/sdk/launch/sweeps/scheduler_sweep.py +91 -0
  359. wandb/sdk/launch/sweeps/utils.py +316 -0
  360. wandb/sdk/launch/utils.py +746 -0
  361. wandb/sdk/launch/wandb_reference.py +138 -0
  362. wandb/sdk/lib/__init__.py +5 -0
  363. wandb/sdk/lib/_settings_toposort_generate.py +159 -0
  364. wandb/sdk/lib/_settings_toposort_generated.py +250 -0
  365. wandb/sdk/lib/_wburls_generate.py +25 -0
  366. wandb/sdk/lib/_wburls_generated.py +22 -0
  367. wandb/sdk/lib/apikey.py +273 -0
  368. wandb/sdk/lib/capped_dict.py +26 -0
  369. wandb/sdk/lib/config_util.py +101 -0
  370. wandb/sdk/lib/credentials.py +141 -0
  371. wandb/sdk/lib/deprecate.py +42 -0
  372. wandb/sdk/lib/disabled.py +29 -0
  373. wandb/sdk/lib/exit_hooks.py +54 -0
  374. wandb/sdk/lib/file_stream_utils.py +118 -0
  375. wandb/sdk/lib/filenames.py +64 -0
  376. wandb/sdk/lib/filesystem.py +372 -0
  377. wandb/sdk/lib/fsm.py +174 -0
  378. wandb/sdk/lib/gitlib.py +239 -0
  379. wandb/sdk/lib/gql_request.py +65 -0
  380. wandb/sdk/lib/handler_util.py +21 -0
  381. wandb/sdk/lib/hashutil.py +84 -0
  382. wandb/sdk/lib/import_hooks.py +275 -0
  383. wandb/sdk/lib/ipython.py +146 -0
  384. wandb/sdk/lib/json_util.py +80 -0
  385. wandb/sdk/lib/lazyloader.py +63 -0
  386. wandb/sdk/lib/mailbox.py +460 -0
  387. wandb/sdk/lib/module.py +69 -0
  388. wandb/sdk/lib/paths.py +106 -0
  389. wandb/sdk/lib/preinit.py +42 -0
  390. wandb/sdk/lib/printer.py +313 -0
  391. wandb/sdk/lib/proto_util.py +90 -0
  392. wandb/sdk/lib/redirect.py +845 -0
  393. wandb/sdk/lib/reporting.py +99 -0
  394. wandb/sdk/lib/retry.py +289 -0
  395. wandb/sdk/lib/run_moment.py +78 -0
  396. wandb/sdk/lib/runid.py +12 -0
  397. wandb/sdk/lib/server.py +52 -0
  398. wandb/sdk/lib/service_connection.py +216 -0
  399. wandb/sdk/lib/service_token.py +94 -0
  400. wandb/sdk/lib/sock_client.py +295 -0
  401. wandb/sdk/lib/sparkline.py +45 -0
  402. wandb/sdk/lib/telemetry.py +100 -0
  403. wandb/sdk/lib/timed_input.py +133 -0
  404. wandb/sdk/lib/timer.py +19 -0
  405. wandb/sdk/lib/tracelog.py +255 -0
  406. wandb/sdk/lib/wburls.py +46 -0
  407. wandb/sdk/service/__init__.py +0 -0
  408. wandb/sdk/service/_startup_debug.py +22 -0
  409. wandb/sdk/service/port_file.py +53 -0
  410. wandb/sdk/service/server.py +116 -0
  411. wandb/sdk/service/server_sock.py +276 -0
  412. wandb/sdk/service/service.py +242 -0
  413. wandb/sdk/service/streams.py +417 -0
  414. wandb/sdk/verify/__init__.py +0 -0
  415. wandb/sdk/verify/verify.py +501 -0
  416. wandb/sdk/wandb_alerts.py +12 -0
  417. wandb/sdk/wandb_config.py +322 -0
  418. wandb/sdk/wandb_helper.py +54 -0
  419. wandb/sdk/wandb_init.py +1266 -0
  420. wandb/sdk/wandb_login.py +349 -0
  421. wandb/sdk/wandb_metric.py +110 -0
  422. wandb/sdk/wandb_require.py +97 -0
  423. wandb/sdk/wandb_require_helpers.py +44 -0
  424. wandb/sdk/wandb_run.py +4236 -0
  425. wandb/sdk/wandb_settings.py +2001 -0
  426. wandb/sdk/wandb_setup.py +409 -0
  427. wandb/sdk/wandb_summary.py +150 -0
  428. wandb/sdk/wandb_sweep.py +119 -0
  429. wandb/sdk/wandb_sync.py +81 -0
  430. wandb/sdk/wandb_watch.py +144 -0
  431. wandb/sklearn.py +35 -0
  432. wandb/sync/__init__.py +3 -0
  433. wandb/sync/sync.py +443 -0
  434. wandb/trigger.py +29 -0
  435. wandb/util.py +1956 -0
  436. wandb/vendor/__init__.py +0 -0
  437. wandb/vendor/gql-0.2.0/setup.py +40 -0
  438. wandb/vendor/gql-0.2.0/tests/__init__.py +0 -0
  439. wandb/vendor/gql-0.2.0/tests/starwars/__init__.py +0 -0
  440. wandb/vendor/gql-0.2.0/tests/starwars/fixtures.py +96 -0
  441. wandb/vendor/gql-0.2.0/tests/starwars/schema.py +146 -0
  442. wandb/vendor/gql-0.2.0/tests/starwars/test_dsl.py +293 -0
  443. wandb/vendor/gql-0.2.0/tests/starwars/test_query.py +355 -0
  444. wandb/vendor/gql-0.2.0/tests/starwars/test_validation.py +171 -0
  445. wandb/vendor/gql-0.2.0/tests/test_client.py +31 -0
  446. wandb/vendor/gql-0.2.0/tests/test_transport.py +89 -0
  447. wandb/vendor/gql-0.2.0/wandb_gql/__init__.py +4 -0
  448. wandb/vendor/gql-0.2.0/wandb_gql/client.py +75 -0
  449. wandb/vendor/gql-0.2.0/wandb_gql/dsl.py +152 -0
  450. wandb/vendor/gql-0.2.0/wandb_gql/gql.py +10 -0
  451. wandb/vendor/gql-0.2.0/wandb_gql/transport/__init__.py +0 -0
  452. wandb/vendor/gql-0.2.0/wandb_gql/transport/http.py +6 -0
  453. wandb/vendor/gql-0.2.0/wandb_gql/transport/local_schema.py +15 -0
  454. wandb/vendor/gql-0.2.0/wandb_gql/transport/requests.py +46 -0
  455. wandb/vendor/gql-0.2.0/wandb_gql/utils.py +21 -0
  456. wandb/vendor/graphql-core-1.1/setup.py +86 -0
  457. wandb/vendor/graphql-core-1.1/wandb_graphql/__init__.py +287 -0
  458. wandb/vendor/graphql-core-1.1/wandb_graphql/error/__init__.py +6 -0
  459. wandb/vendor/graphql-core-1.1/wandb_graphql/error/base.py +42 -0
  460. wandb/vendor/graphql-core-1.1/wandb_graphql/error/format_error.py +11 -0
  461. wandb/vendor/graphql-core-1.1/wandb_graphql/error/located_error.py +29 -0
  462. wandb/vendor/graphql-core-1.1/wandb_graphql/error/syntax_error.py +36 -0
  463. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/__init__.py +26 -0
  464. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/base.py +311 -0
  465. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executor.py +398 -0
  466. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/__init__.py +0 -0
  467. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/asyncio.py +53 -0
  468. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/gevent.py +22 -0
  469. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/process.py +32 -0
  470. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/sync.py +7 -0
  471. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/thread.py +35 -0
  472. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/utils.py +6 -0
  473. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/__init__.py +0 -0
  474. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/executor.py +66 -0
  475. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/fragment.py +252 -0
  476. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/resolver.py +151 -0
  477. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/utils.py +7 -0
  478. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/middleware.py +57 -0
  479. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/values.py +145 -0
  480. wandb/vendor/graphql-core-1.1/wandb_graphql/graphql.py +60 -0
  481. wandb/vendor/graphql-core-1.1/wandb_graphql/language/__init__.py +0 -0
  482. wandb/vendor/graphql-core-1.1/wandb_graphql/language/ast.py +1349 -0
  483. wandb/vendor/graphql-core-1.1/wandb_graphql/language/base.py +19 -0
  484. wandb/vendor/graphql-core-1.1/wandb_graphql/language/lexer.py +435 -0
  485. wandb/vendor/graphql-core-1.1/wandb_graphql/language/location.py +30 -0
  486. wandb/vendor/graphql-core-1.1/wandb_graphql/language/parser.py +779 -0
  487. wandb/vendor/graphql-core-1.1/wandb_graphql/language/printer.py +193 -0
  488. wandb/vendor/graphql-core-1.1/wandb_graphql/language/source.py +18 -0
  489. wandb/vendor/graphql-core-1.1/wandb_graphql/language/visitor.py +222 -0
  490. wandb/vendor/graphql-core-1.1/wandb_graphql/language/visitor_meta.py +82 -0
  491. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/__init__.py +0 -0
  492. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/cached_property.py +17 -0
  493. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/contain_subset.py +28 -0
  494. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/default_ordered_dict.py +40 -0
  495. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/ordereddict.py +8 -0
  496. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/pair_set.py +43 -0
  497. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/version.py +78 -0
  498. wandb/vendor/graphql-core-1.1/wandb_graphql/type/__init__.py +67 -0
  499. wandb/vendor/graphql-core-1.1/wandb_graphql/type/definition.py +619 -0
  500. wandb/vendor/graphql-core-1.1/wandb_graphql/type/directives.py +132 -0
  501. wandb/vendor/graphql-core-1.1/wandb_graphql/type/introspection.py +440 -0
  502. wandb/vendor/graphql-core-1.1/wandb_graphql/type/scalars.py +131 -0
  503. wandb/vendor/graphql-core-1.1/wandb_graphql/type/schema.py +100 -0
  504. wandb/vendor/graphql-core-1.1/wandb_graphql/type/typemap.py +145 -0
  505. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/__init__.py +0 -0
  506. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/assert_valid_name.py +9 -0
  507. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_from_value.py +65 -0
  508. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_to_code.py +49 -0
  509. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_to_dict.py +24 -0
  510. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/base.py +75 -0
  511. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/build_ast_schema.py +291 -0
  512. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/build_client_schema.py +250 -0
  513. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/concat_ast.py +9 -0
  514. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/extend_schema.py +357 -0
  515. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/get_field_def.py +27 -0
  516. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/get_operation_ast.py +21 -0
  517. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/introspection_query.py +90 -0
  518. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/is_valid_literal_value.py +67 -0
  519. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/is_valid_value.py +66 -0
  520. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/quoted_or_list.py +21 -0
  521. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/schema_printer.py +168 -0
  522. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/suggestion_list.py +56 -0
  523. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_comparators.py +69 -0
  524. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_from_ast.py +21 -0
  525. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_info.py +149 -0
  526. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/value_from_ast.py +69 -0
  527. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/__init__.py +4 -0
  528. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/__init__.py +79 -0
  529. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/arguments_of_correct_type.py +24 -0
  530. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/base.py +8 -0
  531. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/default_values_of_correct_type.py +44 -0
  532. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/fields_on_correct_type.py +113 -0
  533. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/fragments_on_composite_types.py +33 -0
  534. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_argument_names.py +70 -0
  535. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_directives.py +97 -0
  536. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_fragment_names.py +19 -0
  537. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_type_names.py +43 -0
  538. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/lone_anonymous_operation.py +23 -0
  539. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_fragment_cycles.py +59 -0
  540. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_undefined_variables.py +36 -0
  541. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_unused_fragments.py +38 -0
  542. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_unused_variables.py +37 -0
  543. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/overlapping_fields_can_be_merged.py +529 -0
  544. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/possible_fragment_spreads.py +44 -0
  545. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/provided_non_null_arguments.py +46 -0
  546. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/scalar_leafs.py +33 -0
  547. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_argument_names.py +32 -0
  548. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_fragment_names.py +28 -0
  549. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_input_field_names.py +33 -0
  550. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_operation_names.py +31 -0
  551. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_variable_names.py +27 -0
  552. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/variables_are_input_types.py +21 -0
  553. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/variables_in_allowed_position.py +53 -0
  554. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/validation.py +158 -0
  555. wandb/vendor/promise-2.3.0/conftest.py +30 -0
  556. wandb/vendor/promise-2.3.0/setup.py +64 -0
  557. wandb/vendor/promise-2.3.0/tests/__init__.py +0 -0
  558. wandb/vendor/promise-2.3.0/tests/conftest.py +8 -0
  559. wandb/vendor/promise-2.3.0/tests/test_awaitable.py +32 -0
  560. wandb/vendor/promise-2.3.0/tests/test_awaitable_35.py +47 -0
  561. wandb/vendor/promise-2.3.0/tests/test_benchmark.py +116 -0
  562. wandb/vendor/promise-2.3.0/tests/test_complex_threads.py +23 -0
  563. wandb/vendor/promise-2.3.0/tests/test_dataloader.py +452 -0
  564. wandb/vendor/promise-2.3.0/tests/test_dataloader_awaitable_35.py +99 -0
  565. wandb/vendor/promise-2.3.0/tests/test_dataloader_extra.py +65 -0
  566. wandb/vendor/promise-2.3.0/tests/test_extra.py +670 -0
  567. wandb/vendor/promise-2.3.0/tests/test_issues.py +132 -0
  568. wandb/vendor/promise-2.3.0/tests/test_promise_list.py +70 -0
  569. wandb/vendor/promise-2.3.0/tests/test_spec.py +584 -0
  570. wandb/vendor/promise-2.3.0/tests/test_thread_safety.py +115 -0
  571. wandb/vendor/promise-2.3.0/tests/utils.py +3 -0
  572. wandb/vendor/promise-2.3.0/wandb_promise/__init__.py +38 -0
  573. wandb/vendor/promise-2.3.0/wandb_promise/async_.py +135 -0
  574. wandb/vendor/promise-2.3.0/wandb_promise/compat.py +32 -0
  575. wandb/vendor/promise-2.3.0/wandb_promise/dataloader.py +326 -0
  576. wandb/vendor/promise-2.3.0/wandb_promise/iterate_promise.py +12 -0
  577. wandb/vendor/promise-2.3.0/wandb_promise/promise.py +848 -0
  578. wandb/vendor/promise-2.3.0/wandb_promise/promise_list.py +151 -0
  579. wandb/vendor/promise-2.3.0/wandb_promise/pyutils/__init__.py +0 -0
  580. wandb/vendor/promise-2.3.0/wandb_promise/pyutils/version.py +83 -0
  581. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/__init__.py +0 -0
  582. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/asyncio.py +22 -0
  583. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/gevent.py +21 -0
  584. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/immediate.py +27 -0
  585. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/thread.py +18 -0
  586. wandb/vendor/promise-2.3.0/wandb_promise/utils.py +56 -0
  587. wandb/vendor/pygments/__init__.py +90 -0
  588. wandb/vendor/pygments/cmdline.py +568 -0
  589. wandb/vendor/pygments/console.py +74 -0
  590. wandb/vendor/pygments/filter.py +74 -0
  591. wandb/vendor/pygments/filters/__init__.py +350 -0
  592. wandb/vendor/pygments/formatter.py +95 -0
  593. wandb/vendor/pygments/formatters/__init__.py +153 -0
  594. wandb/vendor/pygments/formatters/_mapping.py +85 -0
  595. wandb/vendor/pygments/formatters/bbcode.py +109 -0
  596. wandb/vendor/pygments/formatters/html.py +851 -0
  597. wandb/vendor/pygments/formatters/img.py +600 -0
  598. wandb/vendor/pygments/formatters/irc.py +182 -0
  599. wandb/vendor/pygments/formatters/latex.py +482 -0
  600. wandb/vendor/pygments/formatters/other.py +160 -0
  601. wandb/vendor/pygments/formatters/rtf.py +147 -0
  602. wandb/vendor/pygments/formatters/svg.py +153 -0
  603. wandb/vendor/pygments/formatters/terminal.py +136 -0
  604. wandb/vendor/pygments/formatters/terminal256.py +309 -0
  605. wandb/vendor/pygments/lexer.py +871 -0
  606. wandb/vendor/pygments/lexers/__init__.py +329 -0
  607. wandb/vendor/pygments/lexers/_asy_builtins.py +1645 -0
  608. wandb/vendor/pygments/lexers/_cl_builtins.py +232 -0
  609. wandb/vendor/pygments/lexers/_cocoa_builtins.py +72 -0
  610. wandb/vendor/pygments/lexers/_csound_builtins.py +1346 -0
  611. wandb/vendor/pygments/lexers/_lasso_builtins.py +5327 -0
  612. wandb/vendor/pygments/lexers/_lua_builtins.py +295 -0
  613. wandb/vendor/pygments/lexers/_mapping.py +500 -0
  614. wandb/vendor/pygments/lexers/_mql_builtins.py +1172 -0
  615. wandb/vendor/pygments/lexers/_openedge_builtins.py +2547 -0
  616. wandb/vendor/pygments/lexers/_php_builtins.py +4756 -0
  617. wandb/vendor/pygments/lexers/_postgres_builtins.py +621 -0
  618. wandb/vendor/pygments/lexers/_scilab_builtins.py +3094 -0
  619. wandb/vendor/pygments/lexers/_sourcemod_builtins.py +1163 -0
  620. wandb/vendor/pygments/lexers/_stan_builtins.py +532 -0
  621. wandb/vendor/pygments/lexers/_stata_builtins.py +419 -0
  622. wandb/vendor/pygments/lexers/_tsql_builtins.py +1004 -0
  623. wandb/vendor/pygments/lexers/_vim_builtins.py +1939 -0
  624. wandb/vendor/pygments/lexers/actionscript.py +240 -0
  625. wandb/vendor/pygments/lexers/agile.py +24 -0
  626. wandb/vendor/pygments/lexers/algebra.py +221 -0
  627. wandb/vendor/pygments/lexers/ambient.py +76 -0
  628. wandb/vendor/pygments/lexers/ampl.py +87 -0
  629. wandb/vendor/pygments/lexers/apl.py +101 -0
  630. wandb/vendor/pygments/lexers/archetype.py +318 -0
  631. wandb/vendor/pygments/lexers/asm.py +641 -0
  632. wandb/vendor/pygments/lexers/automation.py +374 -0
  633. wandb/vendor/pygments/lexers/basic.py +500 -0
  634. wandb/vendor/pygments/lexers/bibtex.py +160 -0
  635. wandb/vendor/pygments/lexers/business.py +612 -0
  636. wandb/vendor/pygments/lexers/c_cpp.py +252 -0
  637. wandb/vendor/pygments/lexers/c_like.py +541 -0
  638. wandb/vendor/pygments/lexers/capnproto.py +78 -0
  639. wandb/vendor/pygments/lexers/chapel.py +102 -0
  640. wandb/vendor/pygments/lexers/clean.py +288 -0
  641. wandb/vendor/pygments/lexers/compiled.py +34 -0
  642. wandb/vendor/pygments/lexers/configs.py +833 -0
  643. wandb/vendor/pygments/lexers/console.py +114 -0
  644. wandb/vendor/pygments/lexers/crystal.py +393 -0
  645. wandb/vendor/pygments/lexers/csound.py +366 -0
  646. wandb/vendor/pygments/lexers/css.py +689 -0
  647. wandb/vendor/pygments/lexers/d.py +251 -0
  648. wandb/vendor/pygments/lexers/dalvik.py +125 -0
  649. wandb/vendor/pygments/lexers/data.py +555 -0
  650. wandb/vendor/pygments/lexers/diff.py +165 -0
  651. wandb/vendor/pygments/lexers/dotnet.py +691 -0
  652. wandb/vendor/pygments/lexers/dsls.py +878 -0
  653. wandb/vendor/pygments/lexers/dylan.py +289 -0
  654. wandb/vendor/pygments/lexers/ecl.py +125 -0
  655. wandb/vendor/pygments/lexers/eiffel.py +65 -0
  656. wandb/vendor/pygments/lexers/elm.py +121 -0
  657. wandb/vendor/pygments/lexers/erlang.py +533 -0
  658. wandb/vendor/pygments/lexers/esoteric.py +277 -0
  659. wandb/vendor/pygments/lexers/ezhil.py +69 -0
  660. wandb/vendor/pygments/lexers/factor.py +344 -0
  661. wandb/vendor/pygments/lexers/fantom.py +250 -0
  662. wandb/vendor/pygments/lexers/felix.py +273 -0
  663. wandb/vendor/pygments/lexers/forth.py +177 -0
  664. wandb/vendor/pygments/lexers/fortran.py +205 -0
  665. wandb/vendor/pygments/lexers/foxpro.py +428 -0
  666. wandb/vendor/pygments/lexers/functional.py +21 -0
  667. wandb/vendor/pygments/lexers/go.py +101 -0
  668. wandb/vendor/pygments/lexers/grammar_notation.py +213 -0
  669. wandb/vendor/pygments/lexers/graph.py +80 -0
  670. wandb/vendor/pygments/lexers/graphics.py +553 -0
  671. wandb/vendor/pygments/lexers/haskell.py +843 -0
  672. wandb/vendor/pygments/lexers/haxe.py +936 -0
  673. wandb/vendor/pygments/lexers/hdl.py +382 -0
  674. wandb/vendor/pygments/lexers/hexdump.py +103 -0
  675. wandb/vendor/pygments/lexers/html.py +602 -0
  676. wandb/vendor/pygments/lexers/idl.py +270 -0
  677. wandb/vendor/pygments/lexers/igor.py +288 -0
  678. wandb/vendor/pygments/lexers/inferno.py +96 -0
  679. wandb/vendor/pygments/lexers/installers.py +322 -0
  680. wandb/vendor/pygments/lexers/int_fiction.py +1343 -0
  681. wandb/vendor/pygments/lexers/iolang.py +63 -0
  682. wandb/vendor/pygments/lexers/j.py +146 -0
  683. wandb/vendor/pygments/lexers/javascript.py +1525 -0
  684. wandb/vendor/pygments/lexers/julia.py +333 -0
  685. wandb/vendor/pygments/lexers/jvm.py +1573 -0
  686. wandb/vendor/pygments/lexers/lisp.py +2621 -0
  687. wandb/vendor/pygments/lexers/make.py +202 -0
  688. wandb/vendor/pygments/lexers/markup.py +595 -0
  689. wandb/vendor/pygments/lexers/math.py +21 -0
  690. wandb/vendor/pygments/lexers/matlab.py +663 -0
  691. wandb/vendor/pygments/lexers/ml.py +769 -0
  692. wandb/vendor/pygments/lexers/modeling.py +358 -0
  693. wandb/vendor/pygments/lexers/modula2.py +1561 -0
  694. wandb/vendor/pygments/lexers/monte.py +204 -0
  695. wandb/vendor/pygments/lexers/ncl.py +894 -0
  696. wandb/vendor/pygments/lexers/nimrod.py +159 -0
  697. wandb/vendor/pygments/lexers/nit.py +64 -0
  698. wandb/vendor/pygments/lexers/nix.py +136 -0
  699. wandb/vendor/pygments/lexers/oberon.py +105 -0
  700. wandb/vendor/pygments/lexers/objective.py +504 -0
  701. wandb/vendor/pygments/lexers/ooc.py +85 -0
  702. wandb/vendor/pygments/lexers/other.py +41 -0
  703. wandb/vendor/pygments/lexers/parasail.py +79 -0
  704. wandb/vendor/pygments/lexers/parsers.py +835 -0
  705. wandb/vendor/pygments/lexers/pascal.py +644 -0
  706. wandb/vendor/pygments/lexers/pawn.py +199 -0
  707. wandb/vendor/pygments/lexers/perl.py +620 -0
  708. wandb/vendor/pygments/lexers/php.py +267 -0
  709. wandb/vendor/pygments/lexers/praat.py +294 -0
  710. wandb/vendor/pygments/lexers/prolog.py +306 -0
  711. wandb/vendor/pygments/lexers/python.py +939 -0
  712. wandb/vendor/pygments/lexers/qvt.py +152 -0
  713. wandb/vendor/pygments/lexers/r.py +453 -0
  714. wandb/vendor/pygments/lexers/rdf.py +270 -0
  715. wandb/vendor/pygments/lexers/rebol.py +431 -0
  716. wandb/vendor/pygments/lexers/resource.py +85 -0
  717. wandb/vendor/pygments/lexers/rnc.py +67 -0
  718. wandb/vendor/pygments/lexers/roboconf.py +82 -0
  719. wandb/vendor/pygments/lexers/robotframework.py +560 -0
  720. wandb/vendor/pygments/lexers/ruby.py +519 -0
  721. wandb/vendor/pygments/lexers/rust.py +220 -0
  722. wandb/vendor/pygments/lexers/sas.py +228 -0
  723. wandb/vendor/pygments/lexers/scripting.py +1222 -0
  724. wandb/vendor/pygments/lexers/shell.py +794 -0
  725. wandb/vendor/pygments/lexers/smalltalk.py +195 -0
  726. wandb/vendor/pygments/lexers/smv.py +79 -0
  727. wandb/vendor/pygments/lexers/snobol.py +83 -0
  728. wandb/vendor/pygments/lexers/special.py +103 -0
  729. wandb/vendor/pygments/lexers/sql.py +681 -0
  730. wandb/vendor/pygments/lexers/stata.py +108 -0
  731. wandb/vendor/pygments/lexers/supercollider.py +90 -0
  732. wandb/vendor/pygments/lexers/tcl.py +145 -0
  733. wandb/vendor/pygments/lexers/templates.py +2283 -0
  734. wandb/vendor/pygments/lexers/testing.py +207 -0
  735. wandb/vendor/pygments/lexers/text.py +25 -0
  736. wandb/vendor/pygments/lexers/textedit.py +169 -0
  737. wandb/vendor/pygments/lexers/textfmts.py +297 -0
  738. wandb/vendor/pygments/lexers/theorem.py +458 -0
  739. wandb/vendor/pygments/lexers/trafficscript.py +54 -0
  740. wandb/vendor/pygments/lexers/typoscript.py +226 -0
  741. wandb/vendor/pygments/lexers/urbi.py +133 -0
  742. wandb/vendor/pygments/lexers/varnish.py +190 -0
  743. wandb/vendor/pygments/lexers/verification.py +111 -0
  744. wandb/vendor/pygments/lexers/web.py +24 -0
  745. wandb/vendor/pygments/lexers/webmisc.py +988 -0
  746. wandb/vendor/pygments/lexers/whiley.py +116 -0
  747. wandb/vendor/pygments/lexers/x10.py +69 -0
  748. wandb/vendor/pygments/modeline.py +44 -0
  749. wandb/vendor/pygments/plugin.py +68 -0
  750. wandb/vendor/pygments/regexopt.py +92 -0
  751. wandb/vendor/pygments/scanner.py +105 -0
  752. wandb/vendor/pygments/sphinxext.py +158 -0
  753. wandb/vendor/pygments/style.py +155 -0
  754. wandb/vendor/pygments/styles/__init__.py +80 -0
  755. wandb/vendor/pygments/styles/abap.py +29 -0
  756. wandb/vendor/pygments/styles/algol.py +63 -0
  757. wandb/vendor/pygments/styles/algol_nu.py +63 -0
  758. wandb/vendor/pygments/styles/arduino.py +98 -0
  759. wandb/vendor/pygments/styles/autumn.py +65 -0
  760. wandb/vendor/pygments/styles/borland.py +51 -0
  761. wandb/vendor/pygments/styles/bw.py +49 -0
  762. wandb/vendor/pygments/styles/colorful.py +81 -0
  763. wandb/vendor/pygments/styles/default.py +73 -0
  764. wandb/vendor/pygments/styles/emacs.py +72 -0
  765. wandb/vendor/pygments/styles/friendly.py +72 -0
  766. wandb/vendor/pygments/styles/fruity.py +42 -0
  767. wandb/vendor/pygments/styles/igor.py +29 -0
  768. wandb/vendor/pygments/styles/lovelace.py +97 -0
  769. wandb/vendor/pygments/styles/manni.py +75 -0
  770. wandb/vendor/pygments/styles/monokai.py +106 -0
  771. wandb/vendor/pygments/styles/murphy.py +80 -0
  772. wandb/vendor/pygments/styles/native.py +65 -0
  773. wandb/vendor/pygments/styles/paraiso_dark.py +125 -0
  774. wandb/vendor/pygments/styles/paraiso_light.py +125 -0
  775. wandb/vendor/pygments/styles/pastie.py +75 -0
  776. wandb/vendor/pygments/styles/perldoc.py +69 -0
  777. wandb/vendor/pygments/styles/rainbow_dash.py +89 -0
  778. wandb/vendor/pygments/styles/rrt.py +33 -0
  779. wandb/vendor/pygments/styles/sas.py +44 -0
  780. wandb/vendor/pygments/styles/stata.py +40 -0
  781. wandb/vendor/pygments/styles/tango.py +141 -0
  782. wandb/vendor/pygments/styles/trac.py +63 -0
  783. wandb/vendor/pygments/styles/vim.py +63 -0
  784. wandb/vendor/pygments/styles/vs.py +38 -0
  785. wandb/vendor/pygments/styles/xcode.py +51 -0
  786. wandb/vendor/pygments/token.py +213 -0
  787. wandb/vendor/pygments/unistring.py +217 -0
  788. wandb/vendor/pygments/util.py +388 -0
  789. wandb/vendor/pynvml/__init__.py +0 -0
  790. wandb/vendor/pynvml/pynvml.py +4779 -0
  791. wandb/vendor/watchdog_0_9_0/wandb_watchdog/__init__.py +17 -0
  792. wandb/vendor/watchdog_0_9_0/wandb_watchdog/events.py +615 -0
  793. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/__init__.py +98 -0
  794. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/api.py +369 -0
  795. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/fsevents.py +172 -0
  796. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/fsevents2.py +239 -0
  797. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify.py +218 -0
  798. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_buffer.py +81 -0
  799. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_c.py +575 -0
  800. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/kqueue.py +730 -0
  801. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/polling.py +145 -0
  802. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/read_directory_changes.py +133 -0
  803. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/winapi.py +348 -0
  804. wandb/vendor/watchdog_0_9_0/wandb_watchdog/patterns.py +265 -0
  805. wandb/vendor/watchdog_0_9_0/wandb_watchdog/tricks/__init__.py +174 -0
  806. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/__init__.py +151 -0
  807. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/bricks.py +249 -0
  808. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/compat.py +29 -0
  809. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/decorators.py +198 -0
  810. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/delayed_queue.py +88 -0
  811. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/dirsnapshot.py +293 -0
  812. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/echo.py +157 -0
  813. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/event_backport.py +41 -0
  814. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py +40 -0
  815. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py +57 -0
  816. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py +64 -0
  817. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py +123 -0
  818. wandb/vendor/watchdog_0_9_0/wandb_watchdog/version.py +28 -0
  819. wandb/vendor/watchdog_0_9_0/wandb_watchdog/watchmedo.py +577 -0
  820. wandb/wandb_agent.py +588 -0
  821. wandb/wandb_controller.py +721 -0
  822. wandb/wandb_run.py +9 -0
  823. wandb-0.18.2.dist-info/METADATA +213 -0
  824. wandb-0.18.2.dist-info/RECORD +827 -0
  825. wandb-0.18.2.dist-info/WHEEL +5 -0
  826. wandb-0.18.2.dist-info/entry_points.txt +3 -0
  827. wandb-0.18.2.dist-info/licenses/LICENSE +21 -0
@@ -0,0 +1,529 @@
1
+ import itertools
2
+ from collections import OrderedDict
3
+
4
+ from ...error import GraphQLError
5
+ from ...language import ast
6
+ from ...language.printer import print_ast
7
+ from ...pyutils.pair_set import PairSet
8
+ from ...type.definition import (GraphQLInterfaceType, GraphQLList,
9
+ GraphQLNonNull, GraphQLObjectType,
10
+ get_named_type, is_leaf_type)
11
+ from ...utils.type_comparators import is_equal_type
12
+ from ...utils.type_from_ast import type_from_ast
13
+ from .base import ValidationRule
14
+
15
+
16
+ class OverlappingFieldsCanBeMerged(ValidationRule):
17
+ __slots__ = ('_compared_fragments', '_cached_fields_and_fragment_names', )
18
+
19
+ def __init__(self, context):
20
+ super(OverlappingFieldsCanBeMerged, self).__init__(context)
21
+ # A memoization for when two fragments are compared "between" each other for
22
+ # conflicts. Two fragments may be compared many times, so memoizing this can
23
+ # dramatically improve the performance of this validator.
24
+ self._compared_fragments = PairSet()
25
+
26
+ # A cache for the "field map" and list of fragment names found in any given
27
+ # selection set. Selection sets may be asked for this information multiple
28
+ # times, so this improves the performance of this validator.
29
+ self._cached_fields_and_fragment_names = {}
30
+
31
+ def leave_SelectionSet(self, node, key, parent, path, ancestors):
32
+ # Note: we validate on the reverse traversal so deeper conflicts will be
33
+ # caught first, for correct calculation of mutual exclusivity and for
34
+ # clearer error messages.
35
+ # field_map = _collect_field_asts_and_defs(
36
+ # self.context,
37
+ # self.context.get_parent_type(),
38
+ # node
39
+ # )
40
+
41
+ # conflicts = _find_conflicts(self.context, False, field_map, self.compared_set)
42
+ conflicts = _find_conflicts_within_selection_set(self.context, self._cached_fields_and_fragment_names,
43
+ self._compared_fragments, self.context.get_parent_type(),
44
+ node)
45
+
46
+ for (reason_name, reason), fields1, fields2 in conflicts:
47
+ self.context.report_error(GraphQLError(
48
+ self.fields_conflict_message(reason_name, reason),
49
+ list(fields1) + list(fields2)
50
+ ))
51
+
52
+ @staticmethod
53
+ def same_type(type1, type2):
54
+ return is_equal_type(type1, type2)
55
+ # return type1.is_same_type(type2)
56
+
57
+ @classmethod
58
+ def fields_conflict_message(cls, reason_name, reason):
59
+ return (
60
+ 'Fields "{}" conflict because {}. '
61
+ 'Use different aliases on the fields to fetch both if this was '
62
+ 'intentional.'
63
+ ).format(reason_name, cls.reason_message(reason))
64
+
65
+ @classmethod
66
+ def reason_message(cls, reason):
67
+ if isinstance(reason, list):
68
+ return ' and '.join('subfields "{}" conflict because {}'.format(reason_name, cls.reason_message(sub_reason))
69
+ for reason_name, sub_reason in reason)
70
+
71
+ return reason
72
+
73
+
74
+ # Algorithm:
75
+ #
76
+ # Conflicts occur when two fields exist in a query which will produce the same
77
+ # response name, but represent differing values, thus creating a conflict.
78
+ # The algorithm below finds all conflicts via making a series of comparisons
79
+ # between fields. In order to compare as few fields as possible, this makes
80
+ # a series of comparisons "within" sets of fields and "between" sets of fields.
81
+ #
82
+ # Given any selection set, a collection produces both a set of fields by
83
+ # also including all inline fragments, as well as a list of fragments
84
+ # referenced by fragment spreads.
85
+ #
86
+ # A) Each selection set represented in the document first compares "within" its
87
+ # collected set of fields, finding any conflicts between every pair of
88
+ # overlapping fields.
89
+ # Note: This is the only time that a the fields "within" a set are compared
90
+ # to each other. After this only fields "between" sets are compared.
91
+ #
92
+ # B) Also, if any fragment is referenced in a selection set, then a
93
+ # comparison is made "between" the original set of fields and the
94
+ # referenced fragment.
95
+ #
96
+ # C) Also, if multiple fragments are referenced, then comparisons
97
+ # are made "between" each referenced fragment.
98
+ #
99
+ # D) When comparing "between" a set of fields and a referenced fragment, first
100
+ # a comparison is made between each field in the original set of fields and
101
+ # each field in the the referenced set of fields.
102
+ #
103
+ # E) Also, if any fragment is referenced in the referenced selection set,
104
+ # then a comparison is made "between" the original set of fields and the
105
+ # referenced fragment (recursively referring to step D).
106
+ #
107
+ # F) When comparing "between" two fragments, first a comparison is made between
108
+ # each field in the first referenced set of fields and each field in the the
109
+ # second referenced set of fields.
110
+ #
111
+ # G) Also, any fragments referenced by the first must be compared to the
112
+ # second, and any fragments referenced by the second must be compared to the
113
+ # first (recursively referring to step F).
114
+ #
115
+ # H) When comparing two fields, if both have selection sets, then a comparison
116
+ # is made "between" both selection sets, first comparing the set of fields in
117
+ # the first selection set with the set of fields in the second.
118
+ #
119
+ # I) Also, if any fragment is referenced in either selection set, then a
120
+ # comparison is made "between" the other set of fields and the
121
+ # referenced fragment.
122
+ #
123
+ # J) Also, if two fragments are referenced in both selection sets, then a
124
+ # comparison is made "between" the two fragments.
125
+
126
+ def _find_conflicts_within_selection_set(context, cached_fields_and_fragment_names, compared_fragments, parent_type,
127
+ selection_set):
128
+ """Find all conflicts found "within" a selection set, including those found via spreading in fragments.
129
+
130
+ Called when visiting each SelectionSet in the GraphQL Document.
131
+ """
132
+ conflicts = []
133
+
134
+ field_map, fragment_names = _get_fields_and_fragments_names(context, cached_fields_and_fragment_names, parent_type,
135
+ selection_set)
136
+
137
+ # (A) Find all conflicts "within" the fields of this selection set.
138
+ # Note: this is the *only place* `collect_conflicts_within` is called.
139
+ _collect_conflicts_within(
140
+ context,
141
+ conflicts,
142
+ cached_fields_and_fragment_names,
143
+ compared_fragments,
144
+ field_map
145
+ )
146
+
147
+ # (B) Then collect conflicts between these fields and those represented by
148
+ # each spread fragment name found.
149
+ for i, fragment_name in enumerate(fragment_names):
150
+ _collect_conflicts_between_fields_and_fragment(
151
+ context,
152
+ conflicts,
153
+ cached_fields_and_fragment_names,
154
+ compared_fragments,
155
+ False,
156
+ field_map,
157
+ fragment_name,
158
+ )
159
+
160
+ # (C) Then compare this fragment with all other fragments found in this
161
+ # selection set to collect conflicts within fragments spread together.
162
+ # This compares each item in the list of fragment names to every other item
163
+ # in that same list (except for itself).
164
+ for other_fragment_name in fragment_names[i+1:]:
165
+ _collect_conflicts_between_fragments(
166
+ context,
167
+ conflicts,
168
+ cached_fields_and_fragment_names,
169
+ compared_fragments,
170
+ False,
171
+ fragment_name,
172
+ other_fragment_name,
173
+ )
174
+
175
+ return conflicts
176
+
177
+
178
+ def _collect_conflicts_between_fields_and_fragment(context, conflicts, cached_fields_and_fragment_names,
179
+ compared_fragments, are_mutually_exclusive, field_map,
180
+ fragment_name):
181
+
182
+ fragment = context.get_fragment(fragment_name)
183
+
184
+ if not fragment:
185
+ return None
186
+
187
+ field_map2, fragment_names2 = _get_referenced_fields_and_fragment_names(context, cached_fields_and_fragment_names,
188
+ fragment)
189
+
190
+ # (D) First collect any conflicts between the provided collection of fields
191
+ # and the collection of fields represented by the given fragment.
192
+ _collect_conflicts_between(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
193
+ are_mutually_exclusive, field_map, field_map2)
194
+
195
+ # (E) Then collect any conflicts between the provided collection of fields
196
+ # and any fragment names found in the given fragment.
197
+ for fragment_name2 in fragment_names2:
198
+ _collect_conflicts_between_fields_and_fragment(context, conflicts, cached_fields_and_fragment_names,
199
+ compared_fragments, are_mutually_exclusive, field_map,
200
+ fragment_name2)
201
+
202
+
203
+ # Collect all conflicts found between two fragments, including via spreading in
204
+ # any nested fragments
205
+ def _collect_conflicts_between_fragments(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
206
+ are_mutually_exclusive, fragment_name1, fragment_name2):
207
+
208
+ fragment1 = context.get_fragment(fragment_name1)
209
+ fragment2 = context.get_fragment(fragment_name2)
210
+
211
+ if not fragment1 or not fragment2:
212
+ return None
213
+
214
+ # No need to compare a fragment to itself.
215
+ if fragment1 == fragment2:
216
+ return None
217
+
218
+ # Memoize so two fragments are not compared for conflicts more than once.
219
+ if compared_fragments.has(fragment_name1, fragment_name2, are_mutually_exclusive):
220
+ return None
221
+
222
+ compared_fragments.add(fragment_name1, fragment_name2, are_mutually_exclusive)
223
+
224
+ field_map1, fragment_names1 = _get_referenced_fields_and_fragment_names(context, cached_fields_and_fragment_names,
225
+ fragment1)
226
+
227
+ field_map2, fragment_names2 = _get_referenced_fields_and_fragment_names(context, cached_fields_and_fragment_names,
228
+ fragment2)
229
+
230
+ # (F) First, collect all conflicts between these two collections of fields
231
+ # (not including any nested fragments)
232
+ _collect_conflicts_between(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
233
+ are_mutually_exclusive, field_map1, field_map2)
234
+
235
+ # (G) Then collect conflicts between the first fragment and any nested
236
+ # fragments spread in the second fragment.
237
+ for _fragment_name2 in fragment_names2:
238
+ _collect_conflicts_between_fragments(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
239
+ are_mutually_exclusive, fragment_name1, _fragment_name2)
240
+
241
+ # (G) Then collect conflicts between the second fragment and any nested
242
+ # fragments spread in the first fragment.
243
+ for _fragment_name1 in fragment_names1:
244
+ _collect_conflicts_between_fragments(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
245
+ are_mutually_exclusive, _fragment_name1, fragment_name2)
246
+
247
+
248
+ def _find_conflicts_between_sub_selection_sets(context, cached_fields_and_fragment_names, compared_fragments,
249
+ are_mutually_exclusive, parent_type1, selection_set1,
250
+ parent_type2, selection_set2):
251
+ """Find all conflicts found between two selection sets.
252
+
253
+ Includes those found via spreading in fragments. Called when determining if conflicts exist
254
+ between the sub-fields of two overlapping fields.
255
+ """
256
+
257
+ conflicts = []
258
+
259
+ field_map1, fragment_names1 = _get_fields_and_fragments_names(context, cached_fields_and_fragment_names,
260
+ parent_type1, selection_set1)
261
+
262
+ field_map2, fragment_names2 = _get_fields_and_fragments_names(context, cached_fields_and_fragment_names,
263
+ parent_type2, selection_set2)
264
+
265
+ # (H) First, collect all conflicts between these two collections of field.
266
+ _collect_conflicts_between(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
267
+ are_mutually_exclusive, field_map1, field_map2)
268
+
269
+ # (I) Then collect conflicts between the first collection of fields and
270
+ # those referenced by each fragment name associated with the second.
271
+ for fragment_name2 in fragment_names2:
272
+ _collect_conflicts_between_fields_and_fragment(context, conflicts, cached_fields_and_fragment_names,
273
+ compared_fragments, are_mutually_exclusive, field_map1,
274
+ fragment_name2)
275
+
276
+ # (I) Then collect conflicts between the second collection of fields and
277
+ # those referenced by each fragment name associated with the first.
278
+ for fragment_name1 in fragment_names1:
279
+ _collect_conflicts_between_fields_and_fragment(context, conflicts, cached_fields_and_fragment_names,
280
+ compared_fragments, are_mutually_exclusive, field_map2,
281
+ fragment_name1)
282
+
283
+ # (J) Also collect conflicts between any fragment names by the first and
284
+ # fragment names by the second. This compares each item in the first set of
285
+ # names to each item in the second set of names.
286
+ for fragment_name1 in fragment_names1:
287
+ for fragment_name2 in fragment_names2:
288
+ _collect_conflicts_between_fragments(context, conflicts, cached_fields_and_fragment_names,
289
+ compared_fragments, are_mutually_exclusive,
290
+ fragment_name1, fragment_name2)
291
+
292
+ return conflicts
293
+
294
+
295
+ def _collect_conflicts_within(context, conflicts, cached_fields_and_fragment_names, compared_fragments, field_map):
296
+ """Collect all Conflicts "within" one collection of fields."""
297
+
298
+ # field map is a keyed collection, where each key represents a response
299
+ # name and the value at that key is a list of all fields which provide that
300
+ # response name. For every response name, if there are multiple fields, they
301
+ # must be compared to find a potential conflict.
302
+ for response_name, fields in list(field_map.items()):
303
+ # This compares every field in the list to every other field in this list
304
+ # (except to itself). If the list only has one item, nothing needs to
305
+ # be compared.
306
+ for i, field in enumerate(fields):
307
+ for other_field in fields[i+1:]:
308
+ # within one collection is never mutually exclusive
309
+ conflict = _find_conflict(context, cached_fields_and_fragment_names, compared_fragments, False,
310
+ response_name, field, other_field)
311
+ if conflict:
312
+ conflicts.append(conflict)
313
+
314
+
315
+ def _collect_conflicts_between(context, conflicts, cached_fields_and_fragment_names, compared_fragments,
316
+ parent_fields_are_mutually_exclusive, field_map1, field_map2):
317
+ """Collect all Conflicts between two collections of fields.
318
+
319
+ This is similar to, but different from the `collect_conflicts_within` function above. This check assumes that
320
+ `collect_conflicts_within` has already been called on each provided collection of fields.
321
+ This is true because this validator traverses each individual selection set.
322
+ """
323
+ # A field map is a keyed collection, where each key represents a response
324
+ # name and the value at that key is a list of all fields which provide that
325
+ # response name. For any response name which appears in both provided field
326
+ # maps, each field from the first field map must be compared to every field
327
+ # in the second field map to find potential conflicts.
328
+ for response_name, fields1 in list(field_map1.items()):
329
+ fields2 = field_map2.get(response_name)
330
+
331
+ if fields2:
332
+ for field1 in fields1:
333
+ for field2 in fields2:
334
+ conflict = _find_conflict(context, cached_fields_and_fragment_names, compared_fragments,
335
+ parent_fields_are_mutually_exclusive, response_name, field1, field2)
336
+
337
+ if conflict:
338
+ conflicts.append(conflict)
339
+
340
+
341
+ def _find_conflict(context, cached_fields_and_fragment_names, compared_fragments, parent_fields_are_mutually_exclusive,
342
+ response_name, field1, field2):
343
+ """Determines if there is a conflict between two particular fields."""
344
+ parent_type1, ast1, def1 = field1
345
+ parent_type2, ast2, def2 = field2
346
+
347
+ # If it is known that two fields could not possibly apply at the same
348
+ # time, due to the parent types, then it is safe to permit them to diverge
349
+ # in aliased field or arguments used as they will not present any ambiguity
350
+ # by differing.
351
+ # It is known that two parent types could never overlap if they are
352
+ # different Object types. Interface or Union types might overlap - if not
353
+ # in the current state of the schema, then perhaps in some future version,
354
+ # thus may not safely diverge.
355
+
356
+ are_mutually_exclusive = (
357
+ parent_fields_are_mutually_exclusive or (
358
+ parent_type1 != parent_type2 and
359
+ isinstance(parent_type1, GraphQLObjectType) and
360
+ isinstance(parent_type2, GraphQLObjectType)
361
+ )
362
+ )
363
+
364
+ # The return type for each field.
365
+ type1 = def1 and def1.type
366
+ type2 = def2 and def2.type
367
+
368
+ if not are_mutually_exclusive:
369
+ # Two aliases must refer to the same field.
370
+ name1 = ast1.name.value
371
+ name2 = ast2.name.value
372
+
373
+ if name1 != name2:
374
+ return (
375
+ (response_name, '{} and {} are different fields'.format(name1, name2)),
376
+ [ast1],
377
+ [ast2]
378
+ )
379
+
380
+ # Two field calls must have the same arguments.
381
+ if not _same_arguments(ast1.arguments, ast2.arguments):
382
+ return (
383
+ (response_name, 'they have differing arguments'),
384
+ [ast1],
385
+ [ast2]
386
+ )
387
+
388
+ if type1 and type2 and do_types_conflict(type1, type2):
389
+ return (
390
+ (response_name, 'they return conflicting types {} and {}'.format(type1, type2)),
391
+ [ast1],
392
+ [ast2]
393
+ )
394
+
395
+ # Collect and compare sub-fields. Use the same "visited fragment names" list
396
+ # for both collections so fields in a fragment reference are never
397
+ # compared to themselves.
398
+ selection_set1 = ast1.selection_set
399
+ selection_set2 = ast2.selection_set
400
+
401
+ if selection_set1 and selection_set2:
402
+ conflicts = _find_conflicts_between_sub_selection_sets(context, cached_fields_and_fragment_names,
403
+ compared_fragments, are_mutually_exclusive,
404
+ get_named_type(type1), selection_set1,
405
+ get_named_type(type2), selection_set2)
406
+
407
+ return _subfield_conflicts(conflicts, response_name, ast1, ast2)
408
+
409
+
410
+ def _get_fields_and_fragments_names(context, cached_fields_and_fragment_names, parent_type, selection_set):
411
+ cached = cached_fields_and_fragment_names.get(selection_set)
412
+
413
+ if not cached:
414
+ ast_and_defs = OrderedDict()
415
+ fragment_names = OrderedDict()
416
+ _collect_fields_and_fragment_names(context, parent_type, selection_set, ast_and_defs, fragment_names)
417
+ cached = [ast_and_defs, list(fragment_names.keys())]
418
+ cached_fields_and_fragment_names[selection_set] = cached
419
+
420
+ return cached
421
+
422
+
423
+ def _get_referenced_fields_and_fragment_names(context, cached_fields_and_fragment_names, fragment):
424
+ """Given a reference to a fragment, return the represented collection of fields as well as a list of
425
+ nested fragment names referenced via fragment spreads."""
426
+
427
+ # Short-circuit building a type from the AST if possible.
428
+ cached = cached_fields_and_fragment_names.get(fragment.selection_set)
429
+
430
+ if cached:
431
+ return cached
432
+
433
+ fragment_type = type_from_ast(context.get_schema(), fragment.type_condition)
434
+
435
+ return _get_fields_and_fragments_names(context, cached_fields_and_fragment_names,
436
+ fragment_type, fragment.selection_set)
437
+
438
+
439
+ def _collect_fields_and_fragment_names(context, parent_type, selection_set, ast_and_defs, fragment_names):
440
+
441
+ for selection in selection_set.selections:
442
+ if isinstance(selection, ast.Field):
443
+ field_name = selection.name.value
444
+ if isinstance(parent_type, (GraphQLObjectType, GraphQLInterfaceType)):
445
+ field_def = parent_type.fields.get(field_name)
446
+ else:
447
+ field_def = None
448
+
449
+ response_name = selection.alias.value if selection.alias else field_name
450
+
451
+ if not ast_and_defs.get(response_name):
452
+ ast_and_defs[response_name] = []
453
+
454
+ ast_and_defs[response_name].append([parent_type, selection, field_def])
455
+
456
+ elif isinstance(selection, ast.FragmentSpread):
457
+ fragment_names[selection.name.value] = True
458
+ elif isinstance(selection, ast.InlineFragment):
459
+ type_condition = selection.type_condition
460
+ if type_condition:
461
+ inline_fragment_type = type_from_ast(context.get_schema(), selection.type_condition)
462
+ else:
463
+ inline_fragment_type = parent_type
464
+
465
+ _collect_fields_and_fragment_names(context, inline_fragment_type, selection.selection_set, ast_and_defs,
466
+ fragment_names)
467
+
468
+
469
+ def _subfield_conflicts(conflicts, response_name, ast1, ast2):
470
+ """Given a series of Conflicts which occurred between two sub-fields, generate a single Conflict."""
471
+ if conflicts:
472
+ return (
473
+ (response_name, [conflict[0] for conflict in conflicts]),
474
+ tuple(itertools.chain([ast1], *[conflict[1] for conflict in conflicts])),
475
+ tuple(itertools.chain([ast2], *[conflict[2] for conflict in conflicts]))
476
+ )
477
+
478
+
479
+ def do_types_conflict(type1, type2):
480
+ if isinstance(type1, GraphQLList):
481
+ if isinstance(type2, GraphQLList):
482
+ return do_types_conflict(type1.of_type, type2.of_type)
483
+ return True
484
+
485
+ if isinstance(type2, GraphQLList):
486
+ if isinstance(type1, GraphQLList):
487
+ return do_types_conflict(type1.of_type, type2.of_type)
488
+ return True
489
+
490
+ if isinstance(type1, GraphQLNonNull):
491
+ if isinstance(type2, GraphQLNonNull):
492
+ return do_types_conflict(type1.of_type, type2.of_type)
493
+ return True
494
+
495
+ if isinstance(type2, GraphQLNonNull):
496
+ if isinstance(type1, GraphQLNonNull):
497
+ return do_types_conflict(type1.of_type, type2.of_type)
498
+ return True
499
+
500
+ if is_leaf_type(type1) or is_leaf_type(type2):
501
+ return type1 != type2
502
+
503
+ return False
504
+
505
+
506
+ def _same_value(value1, value2):
507
+ return (not value1 and not value2) or print_ast(value1) == print_ast(value2)
508
+
509
+
510
+ def _same_arguments(arguments1, arguments2):
511
+ # Check to see if they are empty arguments or nones. If they are, we can
512
+ # bail out early.
513
+ if not (arguments1 or arguments2):
514
+ return True
515
+
516
+ if len(arguments1) != len(arguments2):
517
+ return False
518
+
519
+ arguments2_values_to_arg = {a.name.value: a for a in arguments2}
520
+
521
+ for argument1 in arguments1:
522
+ argument2 = arguments2_values_to_arg.get(argument1.name.value)
523
+ if not argument2:
524
+ return False
525
+
526
+ if not _same_value(argument1.value, argument2.value):
527
+ return False
528
+
529
+ return True
@@ -0,0 +1,44 @@
1
+ from ...error import GraphQLError
2
+ from ...utils.type_comparators import do_types_overlap
3
+ from ...utils.type_from_ast import type_from_ast
4
+ from .base import ValidationRule
5
+
6
+
7
+ class PossibleFragmentSpreads(ValidationRule):
8
+
9
+ def enter_InlineFragment(self, node, key, parent, path, ancestors):
10
+ frag_type = self.context.get_type()
11
+ parent_type = self.context.get_parent_type()
12
+ schema = self.context.get_schema()
13
+ if frag_type and parent_type and not do_types_overlap(schema, frag_type, parent_type):
14
+ self.context.report_error(GraphQLError(
15
+ self.type_incompatible_anon_spread_message(parent_type, frag_type),
16
+ [node]
17
+ ))
18
+
19
+ def enter_FragmentSpread(self, node, key, parent, path, ancestors):
20
+ frag_name = node.name.value
21
+ frag_type = self.get_fragment_type(self.context, frag_name)
22
+ parent_type = self.context.get_parent_type()
23
+ schema = self.context.get_schema()
24
+ if frag_type and parent_type and not do_types_overlap(schema, frag_type, parent_type):
25
+ self.context.report_error(GraphQLError(
26
+ self.type_incompatible_spread_message(frag_name, parent_type, frag_type),
27
+ [node]
28
+ ))
29
+
30
+ @staticmethod
31
+ def get_fragment_type(context, name):
32
+ frag = context.get_fragment(name)
33
+ return frag and type_from_ast(context.get_schema(), frag.type_condition)
34
+
35
+ @staticmethod
36
+ def type_incompatible_spread_message(frag_name, parent_type, frag_type):
37
+ return 'Fragment {} cannot be spread here as objects of type {} can never be of type {}'.format(frag_name,
38
+ parent_type,
39
+ frag_type)
40
+
41
+ @staticmethod
42
+ def type_incompatible_anon_spread_message(parent_type, frag_type):
43
+ return 'Fragment cannot be spread here as objects of type {} can never be of type {}'.format(parent_type,
44
+ frag_type)
@@ -0,0 +1,46 @@
1
+ from ...error import GraphQLError
2
+ from ...type.definition import GraphQLNonNull
3
+ from .base import ValidationRule
4
+
5
+
6
+ class ProvidedNonNullArguments(ValidationRule):
7
+
8
+ def leave_Field(self, node, key, parent, path, ancestors):
9
+ field_def = self.context.get_field_def()
10
+ if not field_def:
11
+ return False
12
+
13
+ arg_asts = node.arguments or []
14
+ arg_ast_map = {arg.name.value: arg for arg in arg_asts}
15
+
16
+ for arg_name, arg_def in field_def.args.items():
17
+ arg_ast = arg_ast_map.get(arg_name, None)
18
+ if not arg_ast and isinstance(arg_def.type, GraphQLNonNull):
19
+ self.context.report_error(GraphQLError(
20
+ self.missing_field_arg_message(node.name.value, arg_name, arg_def.type),
21
+ [node]
22
+ ))
23
+
24
+ def leave_Directive(self, node, key, parent, path, ancestors):
25
+ directive_def = self.context.get_directive()
26
+ if not directive_def:
27
+ return False
28
+
29
+ arg_asts = node.arguments or []
30
+ arg_ast_map = {arg.name.value: arg for arg in arg_asts}
31
+
32
+ for arg_name, arg_def in directive_def.args.items():
33
+ arg_ast = arg_ast_map.get(arg_name, None)
34
+ if not arg_ast and isinstance(arg_def.type, GraphQLNonNull):
35
+ self.context.report_error(GraphQLError(
36
+ self.missing_directive_arg_message(node.name.value, arg_name, arg_def.type),
37
+ [node]
38
+ ))
39
+
40
+ @staticmethod
41
+ def missing_field_arg_message(name, arg_name, type):
42
+ return 'Field "{}" argument "{}" of type "{}" is required but not provided.'.format(name, arg_name, type)
43
+
44
+ @staticmethod
45
+ def missing_directive_arg_message(name, arg_name, type):
46
+ return 'Directive "{}" argument "{}" of type "{}" is required but not provided.'.format(name, arg_name, type)
@@ -0,0 +1,33 @@
1
+ from ...error import GraphQLError
2
+ from ...type.definition import get_named_type, is_leaf_type
3
+ from .base import ValidationRule
4
+
5
+
6
+ class ScalarLeafs(ValidationRule):
7
+
8
+ def enter_Field(self, node, key, parent, path, ancestors):
9
+ type = self.context.get_type()
10
+
11
+ if not type:
12
+ return
13
+
14
+ if is_leaf_type(get_named_type(type)):
15
+ if node.selection_set:
16
+ self.context.report_error(GraphQLError(
17
+ self.no_subselection_allowed_message(node.name.value, type),
18
+ [node.selection_set]
19
+ ))
20
+
21
+ elif not node.selection_set:
22
+ self.context.report_error(GraphQLError(
23
+ self.required_subselection_message(node.name.value, type),
24
+ [node]
25
+ ))
26
+
27
+ @staticmethod
28
+ def no_subselection_allowed_message(field, type):
29
+ return 'Field "{}" of type "{}" must not have a sub selection.'.format(field, type)
30
+
31
+ @staticmethod
32
+ def required_subselection_message(field, type):
33
+ return 'Field "{}" of type "{}" must have a sub selection.'.format(field, type)
@@ -0,0 +1,32 @@
1
+ from ...error import GraphQLError
2
+ from .base import ValidationRule
3
+
4
+
5
+ class UniqueArgumentNames(ValidationRule):
6
+ __slots__ = 'known_arg_names',
7
+
8
+ def __init__(self, context):
9
+ super(UniqueArgumentNames, self).__init__(context)
10
+ self.known_arg_names = {}
11
+
12
+ def enter_Field(self, node, key, parent, path, ancestors):
13
+ self.known_arg_names = {}
14
+
15
+ def enter_Directive(self, node, key, parent, path, ancestors):
16
+ self.known_arg_names = {}
17
+
18
+ def enter_Argument(self, node, key, parent, path, ancestors):
19
+ arg_name = node.name.value
20
+
21
+ if arg_name in self.known_arg_names:
22
+ self.context.report_error(GraphQLError(
23
+ self.duplicate_arg_message(arg_name),
24
+ [self.known_arg_names[arg_name], node.name]
25
+ ))
26
+ else:
27
+ self.known_arg_names[arg_name] = node.name
28
+ return False
29
+
30
+ @staticmethod
31
+ def duplicate_arg_message(field):
32
+ return 'There can only be one argument named "{}".'.format(field)