wandb 0.19.1__py3-none-musllinux_1_2_aarch64.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (822) hide show
  1. package_readme.md +97 -0
  2. wandb/__init__.py +246 -0
  3. wandb/__init__.pyi +1197 -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 +263 -0
  10. wandb/apis/__init__.py +48 -0
  11. wandb/apis/attrs.py +51 -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 +103 -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 +73 -0
  21. wandb/apis/paginator.py +81 -0
  22. wandb/apis/public/__init__.py +34 -0
  23. wandb/apis/public/api.py +1387 -0
  24. wandb/apis/public/artifacts.py +1095 -0
  25. wandb/apis/public/const.py +4 -0
  26. wandb/apis/public/files.py +263 -0
  27. wandb/apis/public/history.py +149 -0
  28. wandb/apis/public/jobs.py +653 -0
  29. wandb/apis/public/projects.py +154 -0
  30. wandb/apis/public/query_generator.py +166 -0
  31. wandb/apis/public/reports.py +458 -0
  32. wandb/apis/public/runs.py +1012 -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/public/utils.py +68 -0
  37. wandb/apis/reports/__init__.py +1 -0
  38. wandb/apis/reports/v1/__init__.py +8 -0
  39. wandb/apis/reports/v2/__init__.py +8 -0
  40. wandb/apis/workspaces/__init__.py +8 -0
  41. wandb/beta/workflows.py +288 -0
  42. wandb/bin/gpu_stats +0 -0
  43. wandb/bin/wandb-core +0 -0
  44. wandb/cli/__init__.py +0 -0
  45. wandb/cli/beta.py +178 -0
  46. wandb/cli/cli.py +2812 -0
  47. wandb/data_types.py +66 -0
  48. wandb/docker/__init__.py +343 -0
  49. wandb/docker/auth.py +435 -0
  50. wandb/docker/wandb-entrypoint.sh +33 -0
  51. wandb/docker/www_authenticate.py +94 -0
  52. wandb/env.py +513 -0
  53. wandb/errors/__init__.py +17 -0
  54. wandb/errors/errors.py +37 -0
  55. wandb/errors/links.py +73 -0
  56. wandb/errors/term.py +415 -0
  57. wandb/errors/util.py +57 -0
  58. wandb/errors/warnings.py +2 -0
  59. wandb/filesync/__init__.py +0 -0
  60. wandb/filesync/dir_watcher.py +403 -0
  61. wandb/filesync/stats.py +100 -0
  62. wandb/filesync/step_checksum.py +142 -0
  63. wandb/filesync/step_prepare.py +179 -0
  64. wandb/filesync/step_upload.py +287 -0
  65. wandb/filesync/upload_job.py +142 -0
  66. wandb/integration/__init__.py +0 -0
  67. wandb/integration/catboost/__init__.py +5 -0
  68. wandb/integration/catboost/catboost.py +178 -0
  69. wandb/integration/cohere/__init__.py +3 -0
  70. wandb/integration/cohere/cohere.py +21 -0
  71. wandb/integration/cohere/resolver.py +347 -0
  72. wandb/integration/diffusers/__init__.py +3 -0
  73. wandb/integration/diffusers/autologger.py +76 -0
  74. wandb/integration/diffusers/pipeline_resolver.py +50 -0
  75. wandb/integration/diffusers/resolvers/__init__.py +9 -0
  76. wandb/integration/diffusers/resolvers/multimodal.py +882 -0
  77. wandb/integration/diffusers/resolvers/utils.py +102 -0
  78. wandb/integration/fastai/__init__.py +245 -0
  79. wandb/integration/gym/__init__.py +99 -0
  80. wandb/integration/huggingface/__init__.py +3 -0
  81. wandb/integration/huggingface/huggingface.py +18 -0
  82. wandb/integration/huggingface/resolver.py +213 -0
  83. wandb/integration/keras/__init__.py +11 -0
  84. wandb/integration/keras/callbacks/__init__.py +5 -0
  85. wandb/integration/keras/callbacks/metrics_logger.py +129 -0
  86. wandb/integration/keras/callbacks/model_checkpoint.py +188 -0
  87. wandb/integration/keras/callbacks/tables_builder.py +228 -0
  88. wandb/integration/keras/keras.py +1089 -0
  89. wandb/integration/kfp/__init__.py +6 -0
  90. wandb/integration/kfp/helpers.py +28 -0
  91. wandb/integration/kfp/kfp_patch.py +334 -0
  92. wandb/integration/kfp/wandb_logging.py +182 -0
  93. wandb/integration/langchain/__init__.py +3 -0
  94. wandb/integration/langchain/wandb_tracer.py +48 -0
  95. wandb/integration/lightgbm/__init__.py +239 -0
  96. wandb/integration/lightning/__init__.py +0 -0
  97. wandb/integration/lightning/fabric/__init__.py +3 -0
  98. wandb/integration/lightning/fabric/logger.py +764 -0
  99. wandb/integration/metaflow/__init__.py +3 -0
  100. wandb/integration/metaflow/metaflow.py +383 -0
  101. wandb/integration/openai/__init__.py +3 -0
  102. wandb/integration/openai/fine_tuning.py +480 -0
  103. wandb/integration/openai/openai.py +22 -0
  104. wandb/integration/openai/resolver.py +240 -0
  105. wandb/integration/prodigy/__init__.py +3 -0
  106. wandb/integration/prodigy/prodigy.py +299 -0
  107. wandb/integration/sacred/__init__.py +117 -0
  108. wandb/integration/sagemaker/__init__.py +12 -0
  109. wandb/integration/sagemaker/auth.py +28 -0
  110. wandb/integration/sagemaker/config.py +49 -0
  111. wandb/integration/sagemaker/files.py +3 -0
  112. wandb/integration/sagemaker/resources.py +34 -0
  113. wandb/integration/sb3/__init__.py +3 -0
  114. wandb/integration/sb3/sb3.py +147 -0
  115. wandb/integration/sklearn/__init__.py +37 -0
  116. wandb/integration/sklearn/calculate/__init__.py +32 -0
  117. wandb/integration/sklearn/calculate/calibration_curves.py +125 -0
  118. wandb/integration/sklearn/calculate/class_proportions.py +68 -0
  119. wandb/integration/sklearn/calculate/confusion_matrix.py +93 -0
  120. wandb/integration/sklearn/calculate/decision_boundaries.py +40 -0
  121. wandb/integration/sklearn/calculate/elbow_curve.py +55 -0
  122. wandb/integration/sklearn/calculate/feature_importances.py +67 -0
  123. wandb/integration/sklearn/calculate/learning_curve.py +64 -0
  124. wandb/integration/sklearn/calculate/outlier_candidates.py +69 -0
  125. wandb/integration/sklearn/calculate/residuals.py +86 -0
  126. wandb/integration/sklearn/calculate/silhouette.py +118 -0
  127. wandb/integration/sklearn/calculate/summary_metrics.py +62 -0
  128. wandb/integration/sklearn/plot/__init__.py +35 -0
  129. wandb/integration/sklearn/plot/classifier.py +329 -0
  130. wandb/integration/sklearn/plot/clusterer.py +146 -0
  131. wandb/integration/sklearn/plot/regressor.py +121 -0
  132. wandb/integration/sklearn/plot/shared.py +91 -0
  133. wandb/integration/sklearn/utils.py +183 -0
  134. wandb/integration/tensorboard/__init__.py +10 -0
  135. wandb/integration/tensorboard/log.py +354 -0
  136. wandb/integration/tensorboard/monkeypatch.py +186 -0
  137. wandb/integration/tensorflow/__init__.py +5 -0
  138. wandb/integration/tensorflow/estimator_hook.py +54 -0
  139. wandb/integration/torch/__init__.py +0 -0
  140. wandb/integration/torch/wandb_torch.py +554 -0
  141. wandb/integration/ultralytics/__init__.py +11 -0
  142. wandb/integration/ultralytics/bbox_utils.py +215 -0
  143. wandb/integration/ultralytics/callback.py +524 -0
  144. wandb/integration/ultralytics/classification_utils.py +83 -0
  145. wandb/integration/ultralytics/mask_utils.py +202 -0
  146. wandb/integration/ultralytics/pose_utils.py +103 -0
  147. wandb/integration/xgboost/__init__.py +11 -0
  148. wandb/integration/xgboost/xgboost.py +189 -0
  149. wandb/integration/yolov8/__init__.py +0 -0
  150. wandb/integration/yolov8/yolov8.py +284 -0
  151. wandb/jupyter.py +513 -0
  152. wandb/mpmain/__init__.py +0 -0
  153. wandb/mpmain/__main__.py +1 -0
  154. wandb/old/__init__.py +0 -0
  155. wandb/old/core.py +53 -0
  156. wandb/old/settings.py +173 -0
  157. wandb/old/summary.py +440 -0
  158. wandb/plot/__init__.py +28 -0
  159. wandb/plot/bar.py +70 -0
  160. wandb/plot/confusion_matrix.py +181 -0
  161. wandb/plot/custom_chart.py +124 -0
  162. wandb/plot/histogram.py +65 -0
  163. wandb/plot/line.py +74 -0
  164. wandb/plot/line_series.py +176 -0
  165. wandb/plot/pr_curve.py +185 -0
  166. wandb/plot/roc_curve.py +163 -0
  167. wandb/plot/scatter.py +66 -0
  168. wandb/plot/utils.py +183 -0
  169. wandb/plot/viz.py +41 -0
  170. wandb/proto/__init__.py +0 -0
  171. wandb/proto/v3/__init__.py +0 -0
  172. wandb/proto/v3/wandb_base_pb2.py +55 -0
  173. wandb/proto/v3/wandb_internal_pb2.py +1658 -0
  174. wandb/proto/v3/wandb_server_pb2.py +228 -0
  175. wandb/proto/v3/wandb_settings_pb2.py +122 -0
  176. wandb/proto/v3/wandb_telemetry_pb2.py +106 -0
  177. wandb/proto/v4/__init__.py +0 -0
  178. wandb/proto/v4/wandb_base_pb2.py +30 -0
  179. wandb/proto/v4/wandb_internal_pb2.py +370 -0
  180. wandb/proto/v4/wandb_server_pb2.py +67 -0
  181. wandb/proto/v4/wandb_settings_pb2.py +47 -0
  182. wandb/proto/v4/wandb_telemetry_pb2.py +41 -0
  183. wandb/proto/v5/wandb_base_pb2.py +31 -0
  184. wandb/proto/v5/wandb_internal_pb2.py +371 -0
  185. wandb/proto/v5/wandb_server_pb2.py +68 -0
  186. wandb/proto/v5/wandb_settings_pb2.py +48 -0
  187. wandb/proto/v5/wandb_telemetry_pb2.py +42 -0
  188. wandb/proto/wandb_base_pb2.py +10 -0
  189. wandb/proto/wandb_deprecated.py +45 -0
  190. wandb/proto/wandb_generate_deprecated.py +30 -0
  191. wandb/proto/wandb_generate_proto.py +49 -0
  192. wandb/proto/wandb_internal_pb2.py +16 -0
  193. wandb/proto/wandb_server_pb2.py +10 -0
  194. wandb/proto/wandb_settings_pb2.py +10 -0
  195. wandb/proto/wandb_telemetry_pb2.py +10 -0
  196. wandb/py.typed +0 -0
  197. wandb/sdk/__init__.py +37 -0
  198. wandb/sdk/artifacts/__init__.py +0 -0
  199. wandb/sdk/artifacts/_validators.py +121 -0
  200. wandb/sdk/artifacts/artifact.py +2364 -0
  201. wandb/sdk/artifacts/artifact_download_logger.py +43 -0
  202. wandb/sdk/artifacts/artifact_file_cache.py +249 -0
  203. wandb/sdk/artifacts/artifact_instance_cache.py +17 -0
  204. wandb/sdk/artifacts/artifact_manifest.py +75 -0
  205. wandb/sdk/artifacts/artifact_manifest_entry.py +249 -0
  206. wandb/sdk/artifacts/artifact_manifests/__init__.py +0 -0
  207. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +92 -0
  208. wandb/sdk/artifacts/artifact_saver.py +265 -0
  209. wandb/sdk/artifacts/artifact_state.py +11 -0
  210. wandb/sdk/artifacts/artifact_ttl.py +7 -0
  211. wandb/sdk/artifacts/exceptions.py +57 -0
  212. wandb/sdk/artifacts/staging.py +25 -0
  213. wandb/sdk/artifacts/storage_handler.py +62 -0
  214. wandb/sdk/artifacts/storage_handlers/__init__.py +0 -0
  215. wandb/sdk/artifacts/storage_handlers/azure_handler.py +213 -0
  216. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +224 -0
  217. wandb/sdk/artifacts/storage_handlers/http_handler.py +114 -0
  218. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +139 -0
  219. wandb/sdk/artifacts/storage_handlers/multi_handler.py +56 -0
  220. wandb/sdk/artifacts/storage_handlers/s3_handler.py +298 -0
  221. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +72 -0
  222. wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +135 -0
  223. wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +74 -0
  224. wandb/sdk/artifacts/storage_layout.py +6 -0
  225. wandb/sdk/artifacts/storage_policies/__init__.py +4 -0
  226. wandb/sdk/artifacts/storage_policies/register.py +1 -0
  227. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +378 -0
  228. wandb/sdk/artifacts/storage_policy.py +72 -0
  229. wandb/sdk/backend/__init__.py +0 -0
  230. wandb/sdk/backend/backend.py +221 -0
  231. wandb/sdk/data_types/__init__.py +0 -0
  232. wandb/sdk/data_types/_dtypes.py +918 -0
  233. wandb/sdk/data_types/_private.py +10 -0
  234. wandb/sdk/data_types/audio.py +165 -0
  235. wandb/sdk/data_types/base_types/__init__.py +0 -0
  236. wandb/sdk/data_types/base_types/json_metadata.py +55 -0
  237. wandb/sdk/data_types/base_types/media.py +376 -0
  238. wandb/sdk/data_types/base_types/wb_value.py +282 -0
  239. wandb/sdk/data_types/bokeh.py +70 -0
  240. wandb/sdk/data_types/graph.py +405 -0
  241. wandb/sdk/data_types/helper_types/__init__.py +0 -0
  242. wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +305 -0
  243. wandb/sdk/data_types/helper_types/classes.py +159 -0
  244. wandb/sdk/data_types/helper_types/image_mask.py +241 -0
  245. wandb/sdk/data_types/histogram.py +94 -0
  246. wandb/sdk/data_types/html.py +115 -0
  247. wandb/sdk/data_types/image.py +847 -0
  248. wandb/sdk/data_types/molecule.py +241 -0
  249. wandb/sdk/data_types/object_3d.py +470 -0
  250. wandb/sdk/data_types/plotly.py +82 -0
  251. wandb/sdk/data_types/saved_model.py +445 -0
  252. wandb/sdk/data_types/table.py +1204 -0
  253. wandb/sdk/data_types/trace_tree.py +438 -0
  254. wandb/sdk/data_types/utils.py +228 -0
  255. wandb/sdk/data_types/video.py +268 -0
  256. wandb/sdk/integration_utils/__init__.py +0 -0
  257. wandb/sdk/integration_utils/auto_logging.py +232 -0
  258. wandb/sdk/integration_utils/data_logging.py +475 -0
  259. wandb/sdk/interface/__init__.py +0 -0
  260. wandb/sdk/interface/constants.py +4 -0
  261. wandb/sdk/interface/interface.py +1010 -0
  262. wandb/sdk/interface/interface_queue.py +53 -0
  263. wandb/sdk/interface/interface_relay.py +53 -0
  264. wandb/sdk/interface/interface_shared.py +546 -0
  265. wandb/sdk/interface/interface_sock.py +61 -0
  266. wandb/sdk/interface/message_future.py +27 -0
  267. wandb/sdk/interface/message_future_poll.py +50 -0
  268. wandb/sdk/interface/router.py +115 -0
  269. wandb/sdk/interface/router_queue.py +41 -0
  270. wandb/sdk/interface/router_relay.py +37 -0
  271. wandb/sdk/interface/router_sock.py +36 -0
  272. wandb/sdk/interface/summary_record.py +67 -0
  273. wandb/sdk/internal/__init__.py +0 -0
  274. wandb/sdk/internal/context.py +89 -0
  275. wandb/sdk/internal/datastore.py +297 -0
  276. wandb/sdk/internal/file_pusher.py +181 -0
  277. wandb/sdk/internal/file_stream.py +695 -0
  278. wandb/sdk/internal/flow_control.py +263 -0
  279. wandb/sdk/internal/handler.py +905 -0
  280. wandb/sdk/internal/internal.py +403 -0
  281. wandb/sdk/internal/internal_api.py +4587 -0
  282. wandb/sdk/internal/internal_util.py +97 -0
  283. wandb/sdk/internal/job_builder.py +638 -0
  284. wandb/sdk/internal/profiler.py +78 -0
  285. wandb/sdk/internal/progress.py +79 -0
  286. wandb/sdk/internal/run.py +25 -0
  287. wandb/sdk/internal/sample.py +70 -0
  288. wandb/sdk/internal/sender.py +1696 -0
  289. wandb/sdk/internal/sender_config.py +197 -0
  290. wandb/sdk/internal/settings_static.py +97 -0
  291. wandb/sdk/internal/system/__init__.py +0 -0
  292. wandb/sdk/internal/system/assets/__init__.py +25 -0
  293. wandb/sdk/internal/system/assets/aggregators.py +37 -0
  294. wandb/sdk/internal/system/assets/asset_registry.py +20 -0
  295. wandb/sdk/internal/system/assets/cpu.py +163 -0
  296. wandb/sdk/internal/system/assets/disk.py +210 -0
  297. wandb/sdk/internal/system/assets/gpu.py +416 -0
  298. wandb/sdk/internal/system/assets/gpu_amd.py +233 -0
  299. wandb/sdk/internal/system/assets/interfaces.py +205 -0
  300. wandb/sdk/internal/system/assets/ipu.py +177 -0
  301. wandb/sdk/internal/system/assets/memory.py +166 -0
  302. wandb/sdk/internal/system/assets/network.py +125 -0
  303. wandb/sdk/internal/system/assets/open_metrics.py +293 -0
  304. wandb/sdk/internal/system/assets/tpu.py +154 -0
  305. wandb/sdk/internal/system/assets/trainium.py +393 -0
  306. wandb/sdk/internal/system/env_probe_helpers.py +13 -0
  307. wandb/sdk/internal/system/system_info.py +250 -0
  308. wandb/sdk/internal/system/system_monitor.py +222 -0
  309. wandb/sdk/internal/tb_watcher.py +519 -0
  310. wandb/sdk/internal/thread_local_settings.py +18 -0
  311. wandb/sdk/internal/writer.py +204 -0
  312. wandb/sdk/launch/__init__.py +15 -0
  313. wandb/sdk/launch/_launch.py +331 -0
  314. wandb/sdk/launch/_launch_add.py +255 -0
  315. wandb/sdk/launch/_project_spec.py +566 -0
  316. wandb/sdk/launch/agent/__init__.py +5 -0
  317. wandb/sdk/launch/agent/agent.py +924 -0
  318. wandb/sdk/launch/agent/config.py +296 -0
  319. wandb/sdk/launch/agent/job_status_tracker.py +53 -0
  320. wandb/sdk/launch/agent/run_queue_item_file_saver.py +39 -0
  321. wandb/sdk/launch/builder/__init__.py +0 -0
  322. wandb/sdk/launch/builder/abstract.py +156 -0
  323. wandb/sdk/launch/builder/build.py +297 -0
  324. wandb/sdk/launch/builder/context_manager.py +235 -0
  325. wandb/sdk/launch/builder/docker_builder.py +177 -0
  326. wandb/sdk/launch/builder/kaniko_builder.py +594 -0
  327. wandb/sdk/launch/builder/noop.py +58 -0
  328. wandb/sdk/launch/builder/templates/_wandb_bootstrap.py +188 -0
  329. wandb/sdk/launch/builder/templates/dockerfile.py +92 -0
  330. wandb/sdk/launch/create_job.py +528 -0
  331. wandb/sdk/launch/environment/abstract.py +29 -0
  332. wandb/sdk/launch/environment/aws_environment.py +322 -0
  333. wandb/sdk/launch/environment/azure_environment.py +105 -0
  334. wandb/sdk/launch/environment/gcp_environment.py +335 -0
  335. wandb/sdk/launch/environment/local_environment.py +65 -0
  336. wandb/sdk/launch/errors.py +13 -0
  337. wandb/sdk/launch/git_reference.py +109 -0
  338. wandb/sdk/launch/inputs/files.py +148 -0
  339. wandb/sdk/launch/inputs/internal.py +315 -0
  340. wandb/sdk/launch/inputs/manage.py +113 -0
  341. wandb/sdk/launch/inputs/schema.py +39 -0
  342. wandb/sdk/launch/loader.py +249 -0
  343. wandb/sdk/launch/registry/abstract.py +48 -0
  344. wandb/sdk/launch/registry/anon.py +29 -0
  345. wandb/sdk/launch/registry/azure_container_registry.py +124 -0
  346. wandb/sdk/launch/registry/elastic_container_registry.py +192 -0
  347. wandb/sdk/launch/registry/google_artifact_registry.py +219 -0
  348. wandb/sdk/launch/registry/local_registry.py +65 -0
  349. wandb/sdk/launch/runner/__init__.py +0 -0
  350. wandb/sdk/launch/runner/abstract.py +185 -0
  351. wandb/sdk/launch/runner/kubernetes_monitor.py +472 -0
  352. wandb/sdk/launch/runner/kubernetes_runner.py +963 -0
  353. wandb/sdk/launch/runner/local_container.py +301 -0
  354. wandb/sdk/launch/runner/local_process.py +78 -0
  355. wandb/sdk/launch/runner/sagemaker_runner.py +426 -0
  356. wandb/sdk/launch/runner/vertex_runner.py +230 -0
  357. wandb/sdk/launch/sweeps/__init__.py +37 -0
  358. wandb/sdk/launch/sweeps/scheduler.py +740 -0
  359. wandb/sdk/launch/sweeps/scheduler_sweep.py +90 -0
  360. wandb/sdk/launch/sweeps/utils.py +316 -0
  361. wandb/sdk/launch/utils.py +747 -0
  362. wandb/sdk/launch/wandb_reference.py +138 -0
  363. wandb/sdk/lib/__init__.py +5 -0
  364. wandb/sdk/lib/apikey.py +269 -0
  365. wandb/sdk/lib/capped_dict.py +26 -0
  366. wandb/sdk/lib/config_util.py +101 -0
  367. wandb/sdk/lib/credentials.py +141 -0
  368. wandb/sdk/lib/deprecate.py +42 -0
  369. wandb/sdk/lib/disabled.py +29 -0
  370. wandb/sdk/lib/exit_hooks.py +54 -0
  371. wandb/sdk/lib/file_stream_utils.py +118 -0
  372. wandb/sdk/lib/filenames.py +64 -0
  373. wandb/sdk/lib/filesystem.py +372 -0
  374. wandb/sdk/lib/fsm.py +180 -0
  375. wandb/sdk/lib/gitlib.py +239 -0
  376. wandb/sdk/lib/gql_request.py +65 -0
  377. wandb/sdk/lib/handler_util.py +21 -0
  378. wandb/sdk/lib/hashutil.py +84 -0
  379. wandb/sdk/lib/import_hooks.py +275 -0
  380. wandb/sdk/lib/ipython.py +126 -0
  381. wandb/sdk/lib/json_util.py +80 -0
  382. wandb/sdk/lib/lazyloader.py +63 -0
  383. wandb/sdk/lib/mailbox.py +456 -0
  384. wandb/sdk/lib/module.py +78 -0
  385. wandb/sdk/lib/paths.py +106 -0
  386. wandb/sdk/lib/preinit.py +42 -0
  387. wandb/sdk/lib/printer.py +548 -0
  388. wandb/sdk/lib/progress.py +279 -0
  389. wandb/sdk/lib/proto_util.py +90 -0
  390. wandb/sdk/lib/redirect.py +845 -0
  391. wandb/sdk/lib/retry.py +289 -0
  392. wandb/sdk/lib/run_moment.py +72 -0
  393. wandb/sdk/lib/runid.py +12 -0
  394. wandb/sdk/lib/server.py +38 -0
  395. wandb/sdk/lib/service_connection.py +216 -0
  396. wandb/sdk/lib/service_token.py +94 -0
  397. wandb/sdk/lib/sock_client.py +290 -0
  398. wandb/sdk/lib/sparkline.py +44 -0
  399. wandb/sdk/lib/telemetry.py +100 -0
  400. wandb/sdk/lib/timed_input.py +133 -0
  401. wandb/sdk/lib/timer.py +19 -0
  402. wandb/sdk/service/__init__.py +0 -0
  403. wandb/sdk/service/_startup_debug.py +22 -0
  404. wandb/sdk/service/port_file.py +53 -0
  405. wandb/sdk/service/server.py +107 -0
  406. wandb/sdk/service/server_sock.py +274 -0
  407. wandb/sdk/service/service.py +242 -0
  408. wandb/sdk/service/streams.py +425 -0
  409. wandb/sdk/verify/__init__.py +0 -0
  410. wandb/sdk/verify/verify.py +501 -0
  411. wandb/sdk/wandb_alerts.py +12 -0
  412. wandb/sdk/wandb_config.py +322 -0
  413. wandb/sdk/wandb_helper.py +54 -0
  414. wandb/sdk/wandb_init.py +1313 -0
  415. wandb/sdk/wandb_login.py +339 -0
  416. wandb/sdk/wandb_metric.py +110 -0
  417. wandb/sdk/wandb_require.py +94 -0
  418. wandb/sdk/wandb_require_helpers.py +44 -0
  419. wandb/sdk/wandb_run.py +4066 -0
  420. wandb/sdk/wandb_settings.py +1309 -0
  421. wandb/sdk/wandb_setup.py +402 -0
  422. wandb/sdk/wandb_summary.py +150 -0
  423. wandb/sdk/wandb_sweep.py +119 -0
  424. wandb/sdk/wandb_sync.py +82 -0
  425. wandb/sdk/wandb_watch.py +150 -0
  426. wandb/sklearn.py +35 -0
  427. wandb/sync/__init__.py +3 -0
  428. wandb/sync/sync.py +442 -0
  429. wandb/trigger.py +29 -0
  430. wandb/util.py +1955 -0
  431. wandb/vendor/__init__.py +0 -0
  432. wandb/vendor/gql-0.2.0/setup.py +40 -0
  433. wandb/vendor/gql-0.2.0/tests/__init__.py +0 -0
  434. wandb/vendor/gql-0.2.0/tests/starwars/__init__.py +0 -0
  435. wandb/vendor/gql-0.2.0/tests/starwars/fixtures.py +96 -0
  436. wandb/vendor/gql-0.2.0/tests/starwars/schema.py +146 -0
  437. wandb/vendor/gql-0.2.0/tests/starwars/test_dsl.py +293 -0
  438. wandb/vendor/gql-0.2.0/tests/starwars/test_query.py +355 -0
  439. wandb/vendor/gql-0.2.0/tests/starwars/test_validation.py +171 -0
  440. wandb/vendor/gql-0.2.0/tests/test_client.py +31 -0
  441. wandb/vendor/gql-0.2.0/tests/test_transport.py +89 -0
  442. wandb/vendor/gql-0.2.0/wandb_gql/__init__.py +4 -0
  443. wandb/vendor/gql-0.2.0/wandb_gql/client.py +75 -0
  444. wandb/vendor/gql-0.2.0/wandb_gql/dsl.py +152 -0
  445. wandb/vendor/gql-0.2.0/wandb_gql/gql.py +10 -0
  446. wandb/vendor/gql-0.2.0/wandb_gql/transport/__init__.py +0 -0
  447. wandb/vendor/gql-0.2.0/wandb_gql/transport/http.py +6 -0
  448. wandb/vendor/gql-0.2.0/wandb_gql/transport/local_schema.py +15 -0
  449. wandb/vendor/gql-0.2.0/wandb_gql/transport/requests.py +46 -0
  450. wandb/vendor/gql-0.2.0/wandb_gql/utils.py +21 -0
  451. wandb/vendor/graphql-core-1.1/setup.py +86 -0
  452. wandb/vendor/graphql-core-1.1/wandb_graphql/__init__.py +287 -0
  453. wandb/vendor/graphql-core-1.1/wandb_graphql/error/__init__.py +6 -0
  454. wandb/vendor/graphql-core-1.1/wandb_graphql/error/base.py +42 -0
  455. wandb/vendor/graphql-core-1.1/wandb_graphql/error/format_error.py +11 -0
  456. wandb/vendor/graphql-core-1.1/wandb_graphql/error/located_error.py +29 -0
  457. wandb/vendor/graphql-core-1.1/wandb_graphql/error/syntax_error.py +36 -0
  458. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/__init__.py +26 -0
  459. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/base.py +311 -0
  460. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executor.py +398 -0
  461. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/__init__.py +0 -0
  462. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/asyncio.py +53 -0
  463. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/gevent.py +22 -0
  464. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/process.py +32 -0
  465. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/sync.py +7 -0
  466. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/thread.py +35 -0
  467. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/utils.py +6 -0
  468. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/__init__.py +0 -0
  469. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/executor.py +66 -0
  470. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/fragment.py +252 -0
  471. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/resolver.py +151 -0
  472. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/utils.py +7 -0
  473. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/middleware.py +57 -0
  474. wandb/vendor/graphql-core-1.1/wandb_graphql/execution/values.py +145 -0
  475. wandb/vendor/graphql-core-1.1/wandb_graphql/graphql.py +60 -0
  476. wandb/vendor/graphql-core-1.1/wandb_graphql/language/__init__.py +0 -0
  477. wandb/vendor/graphql-core-1.1/wandb_graphql/language/ast.py +1349 -0
  478. wandb/vendor/graphql-core-1.1/wandb_graphql/language/base.py +19 -0
  479. wandb/vendor/graphql-core-1.1/wandb_graphql/language/lexer.py +435 -0
  480. wandb/vendor/graphql-core-1.1/wandb_graphql/language/location.py +30 -0
  481. wandb/vendor/graphql-core-1.1/wandb_graphql/language/parser.py +779 -0
  482. wandb/vendor/graphql-core-1.1/wandb_graphql/language/printer.py +193 -0
  483. wandb/vendor/graphql-core-1.1/wandb_graphql/language/source.py +18 -0
  484. wandb/vendor/graphql-core-1.1/wandb_graphql/language/visitor.py +222 -0
  485. wandb/vendor/graphql-core-1.1/wandb_graphql/language/visitor_meta.py +82 -0
  486. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/__init__.py +0 -0
  487. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/cached_property.py +17 -0
  488. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/contain_subset.py +28 -0
  489. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/default_ordered_dict.py +40 -0
  490. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/ordereddict.py +8 -0
  491. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/pair_set.py +43 -0
  492. wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/version.py +78 -0
  493. wandb/vendor/graphql-core-1.1/wandb_graphql/type/__init__.py +67 -0
  494. wandb/vendor/graphql-core-1.1/wandb_graphql/type/definition.py +619 -0
  495. wandb/vendor/graphql-core-1.1/wandb_graphql/type/directives.py +132 -0
  496. wandb/vendor/graphql-core-1.1/wandb_graphql/type/introspection.py +440 -0
  497. wandb/vendor/graphql-core-1.1/wandb_graphql/type/scalars.py +131 -0
  498. wandb/vendor/graphql-core-1.1/wandb_graphql/type/schema.py +100 -0
  499. wandb/vendor/graphql-core-1.1/wandb_graphql/type/typemap.py +145 -0
  500. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/__init__.py +0 -0
  501. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/assert_valid_name.py +9 -0
  502. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_from_value.py +65 -0
  503. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_to_code.py +49 -0
  504. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_to_dict.py +24 -0
  505. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/base.py +75 -0
  506. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/build_ast_schema.py +291 -0
  507. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/build_client_schema.py +250 -0
  508. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/concat_ast.py +9 -0
  509. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/extend_schema.py +357 -0
  510. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/get_field_def.py +27 -0
  511. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/get_operation_ast.py +21 -0
  512. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/introspection_query.py +90 -0
  513. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/is_valid_literal_value.py +67 -0
  514. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/is_valid_value.py +66 -0
  515. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/quoted_or_list.py +21 -0
  516. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/schema_printer.py +168 -0
  517. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/suggestion_list.py +56 -0
  518. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_comparators.py +69 -0
  519. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_from_ast.py +21 -0
  520. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_info.py +149 -0
  521. wandb/vendor/graphql-core-1.1/wandb_graphql/utils/value_from_ast.py +69 -0
  522. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/__init__.py +4 -0
  523. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/__init__.py +79 -0
  524. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/arguments_of_correct_type.py +24 -0
  525. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/base.py +8 -0
  526. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/default_values_of_correct_type.py +44 -0
  527. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/fields_on_correct_type.py +113 -0
  528. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/fragments_on_composite_types.py +33 -0
  529. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_argument_names.py +70 -0
  530. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_directives.py +97 -0
  531. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_fragment_names.py +19 -0
  532. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_type_names.py +43 -0
  533. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/lone_anonymous_operation.py +23 -0
  534. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_fragment_cycles.py +59 -0
  535. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_undefined_variables.py +36 -0
  536. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_unused_fragments.py +38 -0
  537. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_unused_variables.py +37 -0
  538. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/overlapping_fields_can_be_merged.py +529 -0
  539. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/possible_fragment_spreads.py +44 -0
  540. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/provided_non_null_arguments.py +46 -0
  541. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/scalar_leafs.py +33 -0
  542. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_argument_names.py +32 -0
  543. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_fragment_names.py +28 -0
  544. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_input_field_names.py +33 -0
  545. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_operation_names.py +31 -0
  546. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_variable_names.py +27 -0
  547. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/variables_are_input_types.py +21 -0
  548. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/variables_in_allowed_position.py +53 -0
  549. wandb/vendor/graphql-core-1.1/wandb_graphql/validation/validation.py +158 -0
  550. wandb/vendor/promise-2.3.0/conftest.py +30 -0
  551. wandb/vendor/promise-2.3.0/setup.py +64 -0
  552. wandb/vendor/promise-2.3.0/tests/__init__.py +0 -0
  553. wandb/vendor/promise-2.3.0/tests/conftest.py +8 -0
  554. wandb/vendor/promise-2.3.0/tests/test_awaitable.py +32 -0
  555. wandb/vendor/promise-2.3.0/tests/test_awaitable_35.py +47 -0
  556. wandb/vendor/promise-2.3.0/tests/test_benchmark.py +116 -0
  557. wandb/vendor/promise-2.3.0/tests/test_complex_threads.py +23 -0
  558. wandb/vendor/promise-2.3.0/tests/test_dataloader.py +452 -0
  559. wandb/vendor/promise-2.3.0/tests/test_dataloader_awaitable_35.py +99 -0
  560. wandb/vendor/promise-2.3.0/tests/test_dataloader_extra.py +65 -0
  561. wandb/vendor/promise-2.3.0/tests/test_extra.py +670 -0
  562. wandb/vendor/promise-2.3.0/tests/test_issues.py +132 -0
  563. wandb/vendor/promise-2.3.0/tests/test_promise_list.py +70 -0
  564. wandb/vendor/promise-2.3.0/tests/test_spec.py +584 -0
  565. wandb/vendor/promise-2.3.0/tests/test_thread_safety.py +115 -0
  566. wandb/vendor/promise-2.3.0/tests/utils.py +3 -0
  567. wandb/vendor/promise-2.3.0/wandb_promise/__init__.py +38 -0
  568. wandb/vendor/promise-2.3.0/wandb_promise/async_.py +135 -0
  569. wandb/vendor/promise-2.3.0/wandb_promise/compat.py +32 -0
  570. wandb/vendor/promise-2.3.0/wandb_promise/dataloader.py +326 -0
  571. wandb/vendor/promise-2.3.0/wandb_promise/iterate_promise.py +12 -0
  572. wandb/vendor/promise-2.3.0/wandb_promise/promise.py +848 -0
  573. wandb/vendor/promise-2.3.0/wandb_promise/promise_list.py +151 -0
  574. wandb/vendor/promise-2.3.0/wandb_promise/pyutils/__init__.py +0 -0
  575. wandb/vendor/promise-2.3.0/wandb_promise/pyutils/version.py +83 -0
  576. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/__init__.py +0 -0
  577. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/asyncio.py +22 -0
  578. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/gevent.py +21 -0
  579. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/immediate.py +27 -0
  580. wandb/vendor/promise-2.3.0/wandb_promise/schedulers/thread.py +18 -0
  581. wandb/vendor/promise-2.3.0/wandb_promise/utils.py +56 -0
  582. wandb/vendor/pygments/__init__.py +90 -0
  583. wandb/vendor/pygments/cmdline.py +568 -0
  584. wandb/vendor/pygments/console.py +74 -0
  585. wandb/vendor/pygments/filter.py +74 -0
  586. wandb/vendor/pygments/filters/__init__.py +350 -0
  587. wandb/vendor/pygments/formatter.py +95 -0
  588. wandb/vendor/pygments/formatters/__init__.py +153 -0
  589. wandb/vendor/pygments/formatters/_mapping.py +85 -0
  590. wandb/vendor/pygments/formatters/bbcode.py +109 -0
  591. wandb/vendor/pygments/formatters/html.py +851 -0
  592. wandb/vendor/pygments/formatters/img.py +600 -0
  593. wandb/vendor/pygments/formatters/irc.py +182 -0
  594. wandb/vendor/pygments/formatters/latex.py +482 -0
  595. wandb/vendor/pygments/formatters/other.py +160 -0
  596. wandb/vendor/pygments/formatters/rtf.py +147 -0
  597. wandb/vendor/pygments/formatters/svg.py +153 -0
  598. wandb/vendor/pygments/formatters/terminal.py +136 -0
  599. wandb/vendor/pygments/formatters/terminal256.py +309 -0
  600. wandb/vendor/pygments/lexer.py +871 -0
  601. wandb/vendor/pygments/lexers/__init__.py +329 -0
  602. wandb/vendor/pygments/lexers/_asy_builtins.py +1645 -0
  603. wandb/vendor/pygments/lexers/_cl_builtins.py +232 -0
  604. wandb/vendor/pygments/lexers/_cocoa_builtins.py +72 -0
  605. wandb/vendor/pygments/lexers/_csound_builtins.py +1346 -0
  606. wandb/vendor/pygments/lexers/_lasso_builtins.py +5327 -0
  607. wandb/vendor/pygments/lexers/_lua_builtins.py +295 -0
  608. wandb/vendor/pygments/lexers/_mapping.py +500 -0
  609. wandb/vendor/pygments/lexers/_mql_builtins.py +1172 -0
  610. wandb/vendor/pygments/lexers/_openedge_builtins.py +2547 -0
  611. wandb/vendor/pygments/lexers/_php_builtins.py +4756 -0
  612. wandb/vendor/pygments/lexers/_postgres_builtins.py +621 -0
  613. wandb/vendor/pygments/lexers/_scilab_builtins.py +3094 -0
  614. wandb/vendor/pygments/lexers/_sourcemod_builtins.py +1163 -0
  615. wandb/vendor/pygments/lexers/_stan_builtins.py +532 -0
  616. wandb/vendor/pygments/lexers/_stata_builtins.py +419 -0
  617. wandb/vendor/pygments/lexers/_tsql_builtins.py +1004 -0
  618. wandb/vendor/pygments/lexers/_vim_builtins.py +1939 -0
  619. wandb/vendor/pygments/lexers/actionscript.py +240 -0
  620. wandb/vendor/pygments/lexers/agile.py +24 -0
  621. wandb/vendor/pygments/lexers/algebra.py +221 -0
  622. wandb/vendor/pygments/lexers/ambient.py +76 -0
  623. wandb/vendor/pygments/lexers/ampl.py +87 -0
  624. wandb/vendor/pygments/lexers/apl.py +101 -0
  625. wandb/vendor/pygments/lexers/archetype.py +318 -0
  626. wandb/vendor/pygments/lexers/asm.py +641 -0
  627. wandb/vendor/pygments/lexers/automation.py +374 -0
  628. wandb/vendor/pygments/lexers/basic.py +500 -0
  629. wandb/vendor/pygments/lexers/bibtex.py +160 -0
  630. wandb/vendor/pygments/lexers/business.py +612 -0
  631. wandb/vendor/pygments/lexers/c_cpp.py +252 -0
  632. wandb/vendor/pygments/lexers/c_like.py +541 -0
  633. wandb/vendor/pygments/lexers/capnproto.py +78 -0
  634. wandb/vendor/pygments/lexers/chapel.py +102 -0
  635. wandb/vendor/pygments/lexers/clean.py +288 -0
  636. wandb/vendor/pygments/lexers/compiled.py +34 -0
  637. wandb/vendor/pygments/lexers/configs.py +833 -0
  638. wandb/vendor/pygments/lexers/console.py +114 -0
  639. wandb/vendor/pygments/lexers/crystal.py +393 -0
  640. wandb/vendor/pygments/lexers/csound.py +366 -0
  641. wandb/vendor/pygments/lexers/css.py +689 -0
  642. wandb/vendor/pygments/lexers/d.py +251 -0
  643. wandb/vendor/pygments/lexers/dalvik.py +125 -0
  644. wandb/vendor/pygments/lexers/data.py +555 -0
  645. wandb/vendor/pygments/lexers/diff.py +165 -0
  646. wandb/vendor/pygments/lexers/dotnet.py +691 -0
  647. wandb/vendor/pygments/lexers/dsls.py +878 -0
  648. wandb/vendor/pygments/lexers/dylan.py +289 -0
  649. wandb/vendor/pygments/lexers/ecl.py +125 -0
  650. wandb/vendor/pygments/lexers/eiffel.py +65 -0
  651. wandb/vendor/pygments/lexers/elm.py +121 -0
  652. wandb/vendor/pygments/lexers/erlang.py +533 -0
  653. wandb/vendor/pygments/lexers/esoteric.py +277 -0
  654. wandb/vendor/pygments/lexers/ezhil.py +69 -0
  655. wandb/vendor/pygments/lexers/factor.py +344 -0
  656. wandb/vendor/pygments/lexers/fantom.py +250 -0
  657. wandb/vendor/pygments/lexers/felix.py +273 -0
  658. wandb/vendor/pygments/lexers/forth.py +177 -0
  659. wandb/vendor/pygments/lexers/fortran.py +205 -0
  660. wandb/vendor/pygments/lexers/foxpro.py +428 -0
  661. wandb/vendor/pygments/lexers/functional.py +21 -0
  662. wandb/vendor/pygments/lexers/go.py +101 -0
  663. wandb/vendor/pygments/lexers/grammar_notation.py +213 -0
  664. wandb/vendor/pygments/lexers/graph.py +80 -0
  665. wandb/vendor/pygments/lexers/graphics.py +553 -0
  666. wandb/vendor/pygments/lexers/haskell.py +843 -0
  667. wandb/vendor/pygments/lexers/haxe.py +936 -0
  668. wandb/vendor/pygments/lexers/hdl.py +382 -0
  669. wandb/vendor/pygments/lexers/hexdump.py +103 -0
  670. wandb/vendor/pygments/lexers/html.py +602 -0
  671. wandb/vendor/pygments/lexers/idl.py +270 -0
  672. wandb/vendor/pygments/lexers/igor.py +288 -0
  673. wandb/vendor/pygments/lexers/inferno.py +96 -0
  674. wandb/vendor/pygments/lexers/installers.py +322 -0
  675. wandb/vendor/pygments/lexers/int_fiction.py +1343 -0
  676. wandb/vendor/pygments/lexers/iolang.py +63 -0
  677. wandb/vendor/pygments/lexers/j.py +146 -0
  678. wandb/vendor/pygments/lexers/javascript.py +1525 -0
  679. wandb/vendor/pygments/lexers/julia.py +333 -0
  680. wandb/vendor/pygments/lexers/jvm.py +1573 -0
  681. wandb/vendor/pygments/lexers/lisp.py +2621 -0
  682. wandb/vendor/pygments/lexers/make.py +202 -0
  683. wandb/vendor/pygments/lexers/markup.py +595 -0
  684. wandb/vendor/pygments/lexers/math.py +21 -0
  685. wandb/vendor/pygments/lexers/matlab.py +663 -0
  686. wandb/vendor/pygments/lexers/ml.py +769 -0
  687. wandb/vendor/pygments/lexers/modeling.py +358 -0
  688. wandb/vendor/pygments/lexers/modula2.py +1561 -0
  689. wandb/vendor/pygments/lexers/monte.py +204 -0
  690. wandb/vendor/pygments/lexers/ncl.py +894 -0
  691. wandb/vendor/pygments/lexers/nimrod.py +159 -0
  692. wandb/vendor/pygments/lexers/nit.py +64 -0
  693. wandb/vendor/pygments/lexers/nix.py +136 -0
  694. wandb/vendor/pygments/lexers/oberon.py +105 -0
  695. wandb/vendor/pygments/lexers/objective.py +504 -0
  696. wandb/vendor/pygments/lexers/ooc.py +85 -0
  697. wandb/vendor/pygments/lexers/other.py +41 -0
  698. wandb/vendor/pygments/lexers/parasail.py +79 -0
  699. wandb/vendor/pygments/lexers/parsers.py +835 -0
  700. wandb/vendor/pygments/lexers/pascal.py +644 -0
  701. wandb/vendor/pygments/lexers/pawn.py +199 -0
  702. wandb/vendor/pygments/lexers/perl.py +620 -0
  703. wandb/vendor/pygments/lexers/php.py +267 -0
  704. wandb/vendor/pygments/lexers/praat.py +294 -0
  705. wandb/vendor/pygments/lexers/prolog.py +306 -0
  706. wandb/vendor/pygments/lexers/python.py +939 -0
  707. wandb/vendor/pygments/lexers/qvt.py +152 -0
  708. wandb/vendor/pygments/lexers/r.py +453 -0
  709. wandb/vendor/pygments/lexers/rdf.py +270 -0
  710. wandb/vendor/pygments/lexers/rebol.py +431 -0
  711. wandb/vendor/pygments/lexers/resource.py +85 -0
  712. wandb/vendor/pygments/lexers/rnc.py +67 -0
  713. wandb/vendor/pygments/lexers/roboconf.py +82 -0
  714. wandb/vendor/pygments/lexers/robotframework.py +560 -0
  715. wandb/vendor/pygments/lexers/ruby.py +519 -0
  716. wandb/vendor/pygments/lexers/rust.py +220 -0
  717. wandb/vendor/pygments/lexers/sas.py +228 -0
  718. wandb/vendor/pygments/lexers/scripting.py +1222 -0
  719. wandb/vendor/pygments/lexers/shell.py +794 -0
  720. wandb/vendor/pygments/lexers/smalltalk.py +195 -0
  721. wandb/vendor/pygments/lexers/smv.py +79 -0
  722. wandb/vendor/pygments/lexers/snobol.py +83 -0
  723. wandb/vendor/pygments/lexers/special.py +103 -0
  724. wandb/vendor/pygments/lexers/sql.py +681 -0
  725. wandb/vendor/pygments/lexers/stata.py +108 -0
  726. wandb/vendor/pygments/lexers/supercollider.py +90 -0
  727. wandb/vendor/pygments/lexers/tcl.py +145 -0
  728. wandb/vendor/pygments/lexers/templates.py +2283 -0
  729. wandb/vendor/pygments/lexers/testing.py +207 -0
  730. wandb/vendor/pygments/lexers/text.py +25 -0
  731. wandb/vendor/pygments/lexers/textedit.py +169 -0
  732. wandb/vendor/pygments/lexers/textfmts.py +297 -0
  733. wandb/vendor/pygments/lexers/theorem.py +458 -0
  734. wandb/vendor/pygments/lexers/trafficscript.py +54 -0
  735. wandb/vendor/pygments/lexers/typoscript.py +226 -0
  736. wandb/vendor/pygments/lexers/urbi.py +133 -0
  737. wandb/vendor/pygments/lexers/varnish.py +190 -0
  738. wandb/vendor/pygments/lexers/verification.py +111 -0
  739. wandb/vendor/pygments/lexers/web.py +24 -0
  740. wandb/vendor/pygments/lexers/webmisc.py +988 -0
  741. wandb/vendor/pygments/lexers/whiley.py +116 -0
  742. wandb/vendor/pygments/lexers/x10.py +69 -0
  743. wandb/vendor/pygments/modeline.py +44 -0
  744. wandb/vendor/pygments/plugin.py +68 -0
  745. wandb/vendor/pygments/regexopt.py +92 -0
  746. wandb/vendor/pygments/scanner.py +105 -0
  747. wandb/vendor/pygments/sphinxext.py +158 -0
  748. wandb/vendor/pygments/style.py +155 -0
  749. wandb/vendor/pygments/styles/__init__.py +80 -0
  750. wandb/vendor/pygments/styles/abap.py +29 -0
  751. wandb/vendor/pygments/styles/algol.py +63 -0
  752. wandb/vendor/pygments/styles/algol_nu.py +63 -0
  753. wandb/vendor/pygments/styles/arduino.py +98 -0
  754. wandb/vendor/pygments/styles/autumn.py +65 -0
  755. wandb/vendor/pygments/styles/borland.py +51 -0
  756. wandb/vendor/pygments/styles/bw.py +49 -0
  757. wandb/vendor/pygments/styles/colorful.py +81 -0
  758. wandb/vendor/pygments/styles/default.py +73 -0
  759. wandb/vendor/pygments/styles/emacs.py +72 -0
  760. wandb/vendor/pygments/styles/friendly.py +72 -0
  761. wandb/vendor/pygments/styles/fruity.py +42 -0
  762. wandb/vendor/pygments/styles/igor.py +29 -0
  763. wandb/vendor/pygments/styles/lovelace.py +97 -0
  764. wandb/vendor/pygments/styles/manni.py +75 -0
  765. wandb/vendor/pygments/styles/monokai.py +106 -0
  766. wandb/vendor/pygments/styles/murphy.py +80 -0
  767. wandb/vendor/pygments/styles/native.py +65 -0
  768. wandb/vendor/pygments/styles/paraiso_dark.py +125 -0
  769. wandb/vendor/pygments/styles/paraiso_light.py +125 -0
  770. wandb/vendor/pygments/styles/pastie.py +75 -0
  771. wandb/vendor/pygments/styles/perldoc.py +69 -0
  772. wandb/vendor/pygments/styles/rainbow_dash.py +89 -0
  773. wandb/vendor/pygments/styles/rrt.py +33 -0
  774. wandb/vendor/pygments/styles/sas.py +44 -0
  775. wandb/vendor/pygments/styles/stata.py +40 -0
  776. wandb/vendor/pygments/styles/tango.py +141 -0
  777. wandb/vendor/pygments/styles/trac.py +63 -0
  778. wandb/vendor/pygments/styles/vim.py +63 -0
  779. wandb/vendor/pygments/styles/vs.py +38 -0
  780. wandb/vendor/pygments/styles/xcode.py +51 -0
  781. wandb/vendor/pygments/token.py +213 -0
  782. wandb/vendor/pygments/unistring.py +217 -0
  783. wandb/vendor/pygments/util.py +388 -0
  784. wandb/vendor/pynvml/__init__.py +0 -0
  785. wandb/vendor/pynvml/pynvml.py +4779 -0
  786. wandb/vendor/watchdog_0_9_0/wandb_watchdog/__init__.py +17 -0
  787. wandb/vendor/watchdog_0_9_0/wandb_watchdog/events.py +615 -0
  788. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/__init__.py +98 -0
  789. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/api.py +369 -0
  790. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/fsevents.py +172 -0
  791. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/fsevents2.py +239 -0
  792. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify.py +218 -0
  793. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_buffer.py +81 -0
  794. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_c.py +575 -0
  795. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/kqueue.py +730 -0
  796. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/polling.py +145 -0
  797. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/read_directory_changes.py +133 -0
  798. wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/winapi.py +348 -0
  799. wandb/vendor/watchdog_0_9_0/wandb_watchdog/patterns.py +265 -0
  800. wandb/vendor/watchdog_0_9_0/wandb_watchdog/tricks/__init__.py +174 -0
  801. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/__init__.py +151 -0
  802. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/bricks.py +249 -0
  803. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/compat.py +29 -0
  804. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/decorators.py +198 -0
  805. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/delayed_queue.py +88 -0
  806. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/dirsnapshot.py +293 -0
  807. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/echo.py +157 -0
  808. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/event_backport.py +41 -0
  809. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py +40 -0
  810. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py +57 -0
  811. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py +64 -0
  812. wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py +123 -0
  813. wandb/vendor/watchdog_0_9_0/wandb_watchdog/version.py +28 -0
  814. wandb/vendor/watchdog_0_9_0/wandb_watchdog/watchmedo.py +577 -0
  815. wandb/wandb_agent.py +588 -0
  816. wandb/wandb_controller.py +719 -0
  817. wandb/wandb_run.py +9 -0
  818. wandb-0.19.1.dist-info/METADATA +223 -0
  819. wandb-0.19.1.dist-info/RECORD +822 -0
  820. wandb-0.19.1.dist-info/WHEEL +5 -0
  821. wandb-0.19.1.dist-info/entry_points.txt +3 -0
  822. wandb-0.19.1.dist-info/licenses/LICENSE +21 -0
@@ -0,0 +1,871 @@
1
+ # -*- coding: utf-8 -*-
2
+ """
3
+ pygments.lexer
4
+ ~~~~~~~~~~~~~~
5
+
6
+ Base lexer classes.
7
+
8
+ :copyright: Copyright 2006-2017 by the Pygments team, see AUTHORS.
9
+ :license: BSD, see LICENSE for details.
10
+ """
11
+
12
+ from __future__ import print_function
13
+
14
+ import re
15
+ import sys
16
+ import time
17
+
18
+ from pygments.filter import apply_filters, Filter
19
+ from pygments.filters import get_filter_by_name
20
+ from pygments.token import Error, Text, Other, _TokenType
21
+ from pygments.util import get_bool_opt, get_int_opt, get_list_opt, \
22
+ make_analysator, text_type, add_metaclass, iteritems, Future, guess_decode
23
+ from pygments.regexopt import regex_opt
24
+
25
+ __all__ = ['Lexer', 'RegexLexer', 'ExtendedRegexLexer', 'DelegatingLexer',
26
+ 'LexerContext', 'include', 'inherit', 'bygroups', 'using', 'this',
27
+ 'default', 'words']
28
+
29
+
30
+ _encoding_map = [(b'\xef\xbb\xbf', 'utf-8'),
31
+ (b'\xff\xfe\0\0', 'utf-32'),
32
+ (b'\0\0\xfe\xff', 'utf-32be'),
33
+ (b'\xff\xfe', 'utf-16'),
34
+ (b'\xfe\xff', 'utf-16be')]
35
+
36
+ _default_analyse = staticmethod(lambda x: 0.0)
37
+
38
+
39
+ class LexerMeta(type):
40
+ """
41
+ This metaclass automagically converts ``analyse_text`` methods into
42
+ static methods which always return float values.
43
+ """
44
+
45
+ def __new__(mcs, name, bases, d):
46
+ if 'analyse_text' in d:
47
+ d['analyse_text'] = make_analysator(d['analyse_text'])
48
+ return type.__new__(mcs, name, bases, d)
49
+
50
+
51
+ @add_metaclass(LexerMeta)
52
+ class Lexer(object):
53
+ """
54
+ Lexer for a specific language.
55
+
56
+ Basic options recognized:
57
+ ``stripnl``
58
+ Strip leading and trailing newlines from the input (default: True).
59
+ ``stripall``
60
+ Strip all leading and trailing whitespace from the input
61
+ (default: False).
62
+ ``ensurenl``
63
+ Make sure that the input ends with a newline (default: True). This
64
+ is required for some lexers that consume input linewise.
65
+
66
+ .. versionadded:: 1.3
67
+
68
+ ``tabsize``
69
+ If given and greater than 0, expand tabs in the input (default: 0).
70
+ ``encoding``
71
+ If given, must be an encoding name. This encoding will be used to
72
+ convert the input string to Unicode, if it is not already a Unicode
73
+ string (default: ``'guess'``, which uses a simple UTF-8 / Locale /
74
+ Latin1 detection. Can also be ``'chardet'`` to use the chardet
75
+ library, if it is installed.
76
+ ``inencoding``
77
+ Overrides the ``encoding`` if given.
78
+ """
79
+
80
+ #: Name of the lexer
81
+ name = None
82
+
83
+ #: Shortcuts for the lexer
84
+ aliases = []
85
+
86
+ #: File name globs
87
+ filenames = []
88
+
89
+ #: Secondary file name globs
90
+ alias_filenames = []
91
+
92
+ #: MIME types
93
+ mimetypes = []
94
+
95
+ #: Priority, should multiple lexers match and no content is provided
96
+ priority = 0
97
+
98
+ def __init__(self, **options):
99
+ self.options = options
100
+ self.stripnl = get_bool_opt(options, 'stripnl', True)
101
+ self.stripall = get_bool_opt(options, 'stripall', False)
102
+ self.ensurenl = get_bool_opt(options, 'ensurenl', True)
103
+ self.tabsize = get_int_opt(options, 'tabsize', 0)
104
+ self.encoding = options.get('encoding', 'guess')
105
+ self.encoding = options.get('inencoding') or self.encoding
106
+ self.filters = []
107
+ for filter_ in get_list_opt(options, 'filters', ()):
108
+ self.add_filter(filter_)
109
+
110
+ def __repr__(self):
111
+ if self.options:
112
+ return '<pygments.lexers.%s with %r>' % (self.__class__.__name__,
113
+ self.options)
114
+ else:
115
+ return '<pygments.lexers.%s>' % self.__class__.__name__
116
+
117
+ def add_filter(self, filter_, **options):
118
+ """
119
+ Add a new stream filter to this lexer.
120
+ """
121
+ if not isinstance(filter_, Filter):
122
+ filter_ = get_filter_by_name(filter_, **options)
123
+ self.filters.append(filter_)
124
+
125
+ def analyse_text(text):
126
+ """
127
+ Has to return a float between ``0`` and ``1`` that indicates
128
+ if a lexer wants to highlight this text. Used by ``guess_lexer``.
129
+ If this method returns ``0`` it won't highlight it in any case, if
130
+ it returns ``1`` highlighting with this lexer is guaranteed.
131
+
132
+ The `LexerMeta` metaclass automatically wraps this function so
133
+ that it works like a static method (no ``self`` or ``cls``
134
+ parameter) and the return value is automatically converted to
135
+ `float`. If the return value is an object that is boolean `False`
136
+ it's the same as if the return values was ``0.0``.
137
+ """
138
+
139
+ def get_tokens(self, text, unfiltered=False):
140
+ """
141
+ Return an iterable of (tokentype, value) pairs generated from
142
+ `text`. If `unfiltered` is set to `True`, the filtering mechanism
143
+ is bypassed even if filters are defined.
144
+
145
+ Also preprocess the text, i.e. expand tabs and strip it if
146
+ wanted and applies registered filters.
147
+ """
148
+ if not isinstance(text, text_type):
149
+ if self.encoding == 'guess':
150
+ text, _ = guess_decode(text)
151
+ elif self.encoding == 'chardet':
152
+ try:
153
+ import chardet
154
+ except ImportError:
155
+ raise ImportError('To enable chardet encoding guessing, '
156
+ 'please install the chardet library '
157
+ 'from http://chardet.feedparser.org/')
158
+ # check for BOM first
159
+ decoded = None
160
+ for bom, encoding in _encoding_map:
161
+ if text.startswith(bom):
162
+ decoded = text[len(bom):].decode(encoding, 'replace')
163
+ break
164
+ # no BOM found, so use chardet
165
+ if decoded is None:
166
+ enc = chardet.detect(text[:1024]) # Guess using first 1KB
167
+ decoded = text.decode(enc.get('encoding') or 'utf-8',
168
+ 'replace')
169
+ text = decoded
170
+ else:
171
+ text = text.decode(self.encoding)
172
+ if text.startswith(u'\ufeff'):
173
+ text = text[len(u'\ufeff'):]
174
+ else:
175
+ if text.startswith(u'\ufeff'):
176
+ text = text[len(u'\ufeff'):]
177
+
178
+ # text now *is* a unicode string
179
+ text = text.replace('\r\n', '\n')
180
+ text = text.replace('\r', '\n')
181
+ if self.stripall:
182
+ text = text.strip()
183
+ elif self.stripnl:
184
+ text = text.strip('\n')
185
+ if self.tabsize > 0:
186
+ text = text.expandtabs(self.tabsize)
187
+ if self.ensurenl and not text.endswith('\n'):
188
+ text += '\n'
189
+
190
+ def streamer():
191
+ for _, t, v in self.get_tokens_unprocessed(text):
192
+ yield t, v
193
+ stream = streamer()
194
+ if not unfiltered:
195
+ stream = apply_filters(stream, self.filters, self)
196
+ return stream
197
+
198
+ def get_tokens_unprocessed(self, text):
199
+ """
200
+ Return an iterable of (index, tokentype, value) pairs where "index"
201
+ is the starting position of the token within the input text.
202
+
203
+ In subclasses, implement this method as a generator to
204
+ maximize effectiveness.
205
+ """
206
+ raise NotImplementedError
207
+
208
+
209
+ class DelegatingLexer(Lexer):
210
+ """
211
+ This lexer takes two lexer as arguments. A root lexer and
212
+ a language lexer. First everything is scanned using the language
213
+ lexer, afterwards all ``Other`` tokens are lexed using the root
214
+ lexer.
215
+
216
+ The lexers from the ``template`` lexer package use this base lexer.
217
+ """
218
+
219
+ def __init__(self, _root_lexer, _language_lexer, _needle=Other, **options):
220
+ self.root_lexer = _root_lexer(**options)
221
+ self.language_lexer = _language_lexer(**options)
222
+ self.needle = _needle
223
+ Lexer.__init__(self, **options)
224
+
225
+ def get_tokens_unprocessed(self, text):
226
+ buffered = ''
227
+ insertions = []
228
+ lng_buffer = []
229
+ for i, t, v in self.language_lexer.get_tokens_unprocessed(text):
230
+ if t is self.needle:
231
+ if lng_buffer:
232
+ insertions.append((len(buffered), lng_buffer))
233
+ lng_buffer = []
234
+ buffered += v
235
+ else:
236
+ lng_buffer.append((i, t, v))
237
+ if lng_buffer:
238
+ insertions.append((len(buffered), lng_buffer))
239
+ return do_insertions(insertions,
240
+ self.root_lexer.get_tokens_unprocessed(buffered))
241
+
242
+
243
+ # ------------------------------------------------------------------------------
244
+ # RegexLexer and ExtendedRegexLexer
245
+ #
246
+
247
+
248
+ class include(str): # pylint: disable=invalid-name
249
+ """
250
+ Indicates that a state should include rules from another state.
251
+ """
252
+ pass
253
+
254
+
255
+ class _inherit(object):
256
+ """
257
+ Indicates the a state should inherit from its superclass.
258
+ """
259
+ def __repr__(self):
260
+ return 'inherit'
261
+
262
+ inherit = _inherit() # pylint: disable=invalid-name
263
+
264
+
265
+ class combined(tuple): # pylint: disable=invalid-name
266
+ """
267
+ Indicates a state combined from multiple states.
268
+ """
269
+
270
+ def __new__(cls, *args):
271
+ return tuple.__new__(cls, args)
272
+
273
+ def __init__(self, *args):
274
+ # tuple.__init__ doesn't do anything
275
+ pass
276
+
277
+
278
+ class _PseudoMatch(object):
279
+ """
280
+ A pseudo match object constructed from a string.
281
+ """
282
+
283
+ def __init__(self, start, text):
284
+ self._text = text
285
+ self._start = start
286
+
287
+ def start(self, arg=None):
288
+ return self._start
289
+
290
+ def end(self, arg=None):
291
+ return self._start + len(self._text)
292
+
293
+ def group(self, arg=None):
294
+ if arg:
295
+ raise IndexError('No such group')
296
+ return self._text
297
+
298
+ def groups(self):
299
+ return (self._text,)
300
+
301
+ def groupdict(self):
302
+ return {}
303
+
304
+
305
+ def bygroups(*args):
306
+ """
307
+ Callback that yields multiple actions for each group in the match.
308
+ """
309
+ def callback(lexer, match, ctx=None):
310
+ for i, action in enumerate(args):
311
+ if action is None:
312
+ continue
313
+ elif type(action) is _TokenType:
314
+ data = match.group(i + 1)
315
+ if data:
316
+ yield match.start(i + 1), action, data
317
+ else:
318
+ data = match.group(i + 1)
319
+ if data is not None:
320
+ if ctx:
321
+ ctx.pos = match.start(i + 1)
322
+ for item in action(lexer,
323
+ _PseudoMatch(match.start(i + 1), data), ctx):
324
+ if item:
325
+ yield item
326
+ if ctx:
327
+ ctx.pos = match.end()
328
+ return callback
329
+
330
+
331
+ class _This(object):
332
+ """
333
+ Special singleton used for indicating the caller class.
334
+ Used by ``using``.
335
+ """
336
+ this = _This()
337
+
338
+
339
+ def using(_other, **kwargs):
340
+ """
341
+ Callback that processes the match with a different lexer.
342
+
343
+ The keyword arguments are forwarded to the lexer, except `state` which
344
+ is handled separately.
345
+
346
+ `state` specifies the state that the new lexer will start in, and can
347
+ be an enumerable such as ('root', 'inline', 'string') or a simple
348
+ string which is assumed to be on top of the root state.
349
+
350
+ Note: For that to work, `_other` must not be an `ExtendedRegexLexer`.
351
+ """
352
+ gt_kwargs = {}
353
+ if 'state' in kwargs:
354
+ s = kwargs.pop('state')
355
+ if isinstance(s, (list, tuple)):
356
+ gt_kwargs['stack'] = s
357
+ else:
358
+ gt_kwargs['stack'] = ('root', s)
359
+
360
+ if _other is this:
361
+ def callback(lexer, match, ctx=None):
362
+ # if keyword arguments are given the callback
363
+ # function has to create a new lexer instance
364
+ if kwargs:
365
+ # XXX: cache that somehow
366
+ kwargs.update(lexer.options)
367
+ lx = lexer.__class__(**kwargs)
368
+ else:
369
+ lx = lexer
370
+ s = match.start()
371
+ for i, t, v in lx.get_tokens_unprocessed(match.group(), **gt_kwargs):
372
+ yield i + s, t, v
373
+ if ctx:
374
+ ctx.pos = match.end()
375
+ else:
376
+ def callback(lexer, match, ctx=None):
377
+ # XXX: cache that somehow
378
+ kwargs.update(lexer.options)
379
+ lx = _other(**kwargs)
380
+
381
+ s = match.start()
382
+ for i, t, v in lx.get_tokens_unprocessed(match.group(), **gt_kwargs):
383
+ yield i + s, t, v
384
+ if ctx:
385
+ ctx.pos = match.end()
386
+ return callback
387
+
388
+
389
+ class default:
390
+ """
391
+ Indicates a state or state action (e.g. #pop) to apply.
392
+ For example default('#pop') is equivalent to ('', Token, '#pop')
393
+ Note that state tuples may be used as well.
394
+
395
+ .. versionadded:: 2.0
396
+ """
397
+ def __init__(self, state):
398
+ self.state = state
399
+
400
+
401
+ class words(Future):
402
+ """
403
+ Indicates a list of literal words that is transformed into an optimized
404
+ regex that matches any of the words.
405
+
406
+ .. versionadded:: 2.0
407
+ """
408
+ def __init__(self, words, prefix='', suffix=''):
409
+ self.words = words
410
+ self.prefix = prefix
411
+ self.suffix = suffix
412
+
413
+ def get(self):
414
+ return regex_opt(self.words, prefix=self.prefix, suffix=self.suffix)
415
+
416
+
417
+ class RegexLexerMeta(LexerMeta):
418
+ """
419
+ Metaclass for RegexLexer, creates the self._tokens attribute from
420
+ self.tokens on the first instantiation.
421
+ """
422
+
423
+ def _process_regex(cls, regex, rflags, state):
424
+ """Preprocess the regular expression component of a token definition."""
425
+ if isinstance(regex, Future):
426
+ regex = regex.get()
427
+ return re.compile(regex, rflags).match
428
+
429
+ def _process_token(cls, token):
430
+ """Preprocess the token component of a token definition."""
431
+ assert type(token) is _TokenType or callable(token), \
432
+ 'token type must be simple type or callable, not %r' % (token,)
433
+ return token
434
+
435
+ def _process_new_state(cls, new_state, unprocessed, processed):
436
+ """Preprocess the state transition action of a token definition."""
437
+ if isinstance(new_state, str):
438
+ # an existing state
439
+ if new_state == '#pop':
440
+ return -1
441
+ elif new_state in unprocessed:
442
+ return (new_state,)
443
+ elif new_state == '#push':
444
+ return new_state
445
+ elif new_state[:5] == '#pop:':
446
+ return -int(new_state[5:])
447
+ else:
448
+ assert False, 'unknown new state %r' % new_state
449
+ elif isinstance(new_state, combined):
450
+ # combine a new state from existing ones
451
+ tmp_state = '_tmp_%d' % cls._tmpname
452
+ cls._tmpname += 1
453
+ itokens = []
454
+ for istate in new_state:
455
+ assert istate != new_state, 'circular state ref %r' % istate
456
+ itokens.extend(cls._process_state(unprocessed,
457
+ processed, istate))
458
+ processed[tmp_state] = itokens
459
+ return (tmp_state,)
460
+ elif isinstance(new_state, tuple):
461
+ # push more than one state
462
+ for istate in new_state:
463
+ assert (istate in unprocessed or
464
+ istate in ('#pop', '#push')), \
465
+ 'unknown new state ' + istate
466
+ return new_state
467
+ else:
468
+ assert False, 'unknown new state def %r' % new_state
469
+
470
+ def _process_state(cls, unprocessed, processed, state):
471
+ """Preprocess a single state definition."""
472
+ assert type(state) is str, "wrong state name %r" % state
473
+ assert state[0] != '#', "invalid state name %r" % state
474
+ if state in processed:
475
+ return processed[state]
476
+ tokens = processed[state] = []
477
+ rflags = cls.flags
478
+ for tdef in unprocessed[state]:
479
+ if isinstance(tdef, include):
480
+ # it's a state reference
481
+ assert tdef != state, "circular state reference %r" % state
482
+ tokens.extend(cls._process_state(unprocessed, processed,
483
+ str(tdef)))
484
+ continue
485
+ if isinstance(tdef, _inherit):
486
+ # should be processed already, but may not in the case of:
487
+ # 1. the state has no counterpart in any parent
488
+ # 2. the state includes more than one 'inherit'
489
+ continue
490
+ if isinstance(tdef, default):
491
+ new_state = cls._process_new_state(tdef.state, unprocessed, processed)
492
+ tokens.append((re.compile('').match, None, new_state))
493
+ continue
494
+
495
+ assert type(tdef) is tuple, "wrong rule def %r" % tdef
496
+
497
+ try:
498
+ rex = cls._process_regex(tdef[0], rflags, state)
499
+ except Exception as err:
500
+ raise ValueError("uncompilable regex %r in state %r of %r: %s" %
501
+ (tdef[0], state, cls, err))
502
+
503
+ token = cls._process_token(tdef[1])
504
+
505
+ if len(tdef) == 2:
506
+ new_state = None
507
+ else:
508
+ new_state = cls._process_new_state(tdef[2],
509
+ unprocessed, processed)
510
+
511
+ tokens.append((rex, token, new_state))
512
+ return tokens
513
+
514
+ def process_tokendef(cls, name, tokendefs=None):
515
+ """Preprocess a dictionary of token definitions."""
516
+ processed = cls._all_tokens[name] = {}
517
+ tokendefs = tokendefs or cls.tokens[name]
518
+ for state in list(tokendefs):
519
+ cls._process_state(tokendefs, processed, state)
520
+ return processed
521
+
522
+ def get_tokendefs(cls):
523
+ """
524
+ Merge tokens from superclasses in MRO order, returning a single tokendef
525
+ dictionary.
526
+
527
+ Any state that is not defined by a subclass will be inherited
528
+ automatically. States that *are* defined by subclasses will, by
529
+ default, override that state in the superclass. If a subclass wishes to
530
+ inherit definitions from a superclass, it can use the special value
531
+ "inherit", which will cause the superclass' state definition to be
532
+ included at that point in the state.
533
+ """
534
+ tokens = {}
535
+ inheritable = {}
536
+ for c in cls.__mro__:
537
+ toks = c.__dict__.get('tokens', {})
538
+
539
+ for state, items in iteritems(toks):
540
+ curitems = tokens.get(state)
541
+ if curitems is None:
542
+ # N.b. because this is assigned by reference, sufficiently
543
+ # deep hierarchies are processed incrementally (e.g. for
544
+ # A(B), B(C), C(RegexLexer), B will be premodified so X(B)
545
+ # will not see any inherits in B).
546
+ tokens[state] = items
547
+ try:
548
+ inherit_ndx = items.index(inherit)
549
+ except ValueError:
550
+ continue
551
+ inheritable[state] = inherit_ndx
552
+ continue
553
+
554
+ inherit_ndx = inheritable.pop(state, None)
555
+ if inherit_ndx is None:
556
+ continue
557
+
558
+ # Replace the "inherit" value with the items
559
+ curitems[inherit_ndx:inherit_ndx+1] = items
560
+ try:
561
+ # N.b. this is the index in items (that is, the superclass
562
+ # copy), so offset required when storing below.
563
+ new_inh_ndx = items.index(inherit)
564
+ except ValueError:
565
+ pass
566
+ else:
567
+ inheritable[state] = inherit_ndx + new_inh_ndx
568
+
569
+ return tokens
570
+
571
+ def __call__(cls, *args, **kwds):
572
+ """Instantiate cls after preprocessing its token definitions."""
573
+ if '_tokens' not in cls.__dict__:
574
+ cls._all_tokens = {}
575
+ cls._tmpname = 0
576
+ if hasattr(cls, 'token_variants') and cls.token_variants:
577
+ # don't process yet
578
+ pass
579
+ else:
580
+ cls._tokens = cls.process_tokendef('', cls.get_tokendefs())
581
+
582
+ return type.__call__(cls, *args, **kwds)
583
+
584
+
585
+ @add_metaclass(RegexLexerMeta)
586
+ class RegexLexer(Lexer):
587
+ """
588
+ Base for simple stateful regular expression-based lexers.
589
+ Simplifies the lexing process so that you need only
590
+ provide a list of states and regular expressions.
591
+ """
592
+
593
+ #: Flags for compiling the regular expressions.
594
+ #: Defaults to MULTILINE.
595
+ flags = re.MULTILINE
596
+
597
+ #: Dict of ``{'state': [(regex, tokentype, new_state), ...], ...}``
598
+ #:
599
+ #: The initial state is 'root'.
600
+ #: ``new_state`` can be omitted to signify no state transition.
601
+ #: If it is a string, the state is pushed on the stack and changed.
602
+ #: If it is a tuple of strings, all states are pushed on the stack and
603
+ #: the current state will be the topmost.
604
+ #: It can also be ``combined('state1', 'state2', ...)``
605
+ #: to signify a new, anonymous state combined from the rules of two
606
+ #: or more existing ones.
607
+ #: Furthermore, it can be '#pop' to signify going back one step in
608
+ #: the state stack, or '#push' to push the current state on the stack
609
+ #: again.
610
+ #:
611
+ #: The tuple can also be replaced with ``include('state')``, in which
612
+ #: case the rules from the state named by the string are included in the
613
+ #: current one.
614
+ tokens = {}
615
+
616
+ def get_tokens_unprocessed(self, text, stack=('root',)):
617
+ """
618
+ Split ``text`` into (tokentype, text) pairs.
619
+
620
+ ``stack`` is the inital stack (default: ``['root']``)
621
+ """
622
+ pos = 0
623
+ tokendefs = self._tokens
624
+ statestack = list(stack)
625
+ statetokens = tokendefs[statestack[-1]]
626
+ while 1:
627
+ for rexmatch, action, new_state in statetokens:
628
+ m = rexmatch(text, pos)
629
+ if m:
630
+ if action is not None:
631
+ if type(action) is _TokenType:
632
+ yield pos, action, m.group()
633
+ else:
634
+ for item in action(self, m):
635
+ yield item
636
+ pos = m.end()
637
+ if new_state is not None:
638
+ # state transition
639
+ if isinstance(new_state, tuple):
640
+ for state in new_state:
641
+ if state == '#pop':
642
+ statestack.pop()
643
+ elif state == '#push':
644
+ statestack.append(statestack[-1])
645
+ else:
646
+ statestack.append(state)
647
+ elif isinstance(new_state, int):
648
+ # pop
649
+ del statestack[new_state:]
650
+ elif new_state == '#push':
651
+ statestack.append(statestack[-1])
652
+ else:
653
+ assert False, "wrong state def: %r" % new_state
654
+ statetokens = tokendefs[statestack[-1]]
655
+ break
656
+ else:
657
+ # We are here only if all state tokens have been considered
658
+ # and there was not a match on any of them.
659
+ try:
660
+ if text[pos] == '\n':
661
+ # at EOL, reset state to "root"
662
+ statestack = ['root']
663
+ statetokens = tokendefs['root']
664
+ yield pos, Text, u'\n'
665
+ pos += 1
666
+ continue
667
+ yield pos, Error, text[pos]
668
+ pos += 1
669
+ except IndexError:
670
+ break
671
+
672
+
673
+ class LexerContext(object):
674
+ """
675
+ A helper object that holds lexer position data.
676
+ """
677
+
678
+ def __init__(self, text, pos, stack=None, end=None):
679
+ self.text = text
680
+ self.pos = pos
681
+ self.end = end or len(text) # end=0 not supported ;-)
682
+ self.stack = stack or ['root']
683
+
684
+ def __repr__(self):
685
+ return 'LexerContext(%r, %r, %r)' % (
686
+ self.text, self.pos, self.stack)
687
+
688
+
689
+ class ExtendedRegexLexer(RegexLexer):
690
+ """
691
+ A RegexLexer that uses a context object to store its state.
692
+ """
693
+
694
+ def get_tokens_unprocessed(self, text=None, context=None):
695
+ """
696
+ Split ``text`` into (tokentype, text) pairs.
697
+ If ``context`` is given, use this lexer context instead.
698
+ """
699
+ tokendefs = self._tokens
700
+ if not context:
701
+ ctx = LexerContext(text, 0)
702
+ statetokens = tokendefs['root']
703
+ else:
704
+ ctx = context
705
+ statetokens = tokendefs[ctx.stack[-1]]
706
+ text = ctx.text
707
+ while 1:
708
+ for rexmatch, action, new_state in statetokens:
709
+ m = rexmatch(text, ctx.pos, ctx.end)
710
+ if m:
711
+ if action is not None:
712
+ if type(action) is _TokenType:
713
+ yield ctx.pos, action, m.group()
714
+ ctx.pos = m.end()
715
+ else:
716
+ for item in action(self, m, ctx):
717
+ yield item
718
+ if not new_state:
719
+ # altered the state stack?
720
+ statetokens = tokendefs[ctx.stack[-1]]
721
+ # CAUTION: callback must set ctx.pos!
722
+ if new_state is not None:
723
+ # state transition
724
+ if isinstance(new_state, tuple):
725
+ for state in new_state:
726
+ if state == '#pop':
727
+ ctx.stack.pop()
728
+ elif state == '#push':
729
+ ctx.stack.append(ctx.stack[-1])
730
+ else:
731
+ ctx.stack.append(state)
732
+ elif isinstance(new_state, int):
733
+ # pop
734
+ del ctx.stack[new_state:]
735
+ elif new_state == '#push':
736
+ ctx.stack.append(ctx.stack[-1])
737
+ else:
738
+ assert False, "wrong state def: %r" % new_state
739
+ statetokens = tokendefs[ctx.stack[-1]]
740
+ break
741
+ else:
742
+ try:
743
+ if ctx.pos >= ctx.end:
744
+ break
745
+ if text[ctx.pos] == '\n':
746
+ # at EOL, reset state to "root"
747
+ ctx.stack = ['root']
748
+ statetokens = tokendefs['root']
749
+ yield ctx.pos, Text, u'\n'
750
+ ctx.pos += 1
751
+ continue
752
+ yield ctx.pos, Error, text[ctx.pos]
753
+ ctx.pos += 1
754
+ except IndexError:
755
+ break
756
+
757
+
758
+ def do_insertions(insertions, tokens):
759
+ """
760
+ Helper for lexers which must combine the results of several
761
+ sublexers.
762
+
763
+ ``insertions`` is a list of ``(index, itokens)`` pairs.
764
+ Each ``itokens`` iterable should be inserted at position
765
+ ``index`` into the token stream given by the ``tokens``
766
+ argument.
767
+
768
+ The result is a combined token stream.
769
+
770
+ TODO: clean up the code here.
771
+ """
772
+ insertions = iter(insertions)
773
+ try:
774
+ index, itokens = next(insertions)
775
+ except StopIteration:
776
+ # no insertions
777
+ for item in tokens:
778
+ yield item
779
+ return
780
+
781
+ realpos = None
782
+ insleft = True
783
+
784
+ # iterate over the token stream where we want to insert
785
+ # the tokens from the insertion list.
786
+ for i, t, v in tokens:
787
+ # first iteration. store the postition of first item
788
+ if realpos is None:
789
+ realpos = i
790
+ oldi = 0
791
+ while insleft and i + len(v) >= index:
792
+ tmpval = v[oldi:index - i]
793
+ yield realpos, t, tmpval
794
+ realpos += len(tmpval)
795
+ for it_index, it_token, it_value in itokens:
796
+ yield realpos, it_token, it_value
797
+ realpos += len(it_value)
798
+ oldi = index - i
799
+ try:
800
+ index, itokens = next(insertions)
801
+ except StopIteration:
802
+ insleft = False
803
+ break # not strictly necessary
804
+ yield realpos, t, v[oldi:]
805
+ realpos += len(v) - oldi
806
+
807
+ # leftover tokens
808
+ while insleft:
809
+ # no normal tokens, set realpos to zero
810
+ realpos = realpos or 0
811
+ for p, t, v in itokens:
812
+ yield realpos, t, v
813
+ realpos += len(v)
814
+ try:
815
+ index, itokens = next(insertions)
816
+ except StopIteration:
817
+ insleft = False
818
+ break # not strictly necessary
819
+
820
+
821
+ class ProfilingRegexLexerMeta(RegexLexerMeta):
822
+ """Metaclass for ProfilingRegexLexer, collects regex timing info."""
823
+
824
+ def _process_regex(cls, regex, rflags, state):
825
+ if isinstance(regex, words):
826
+ rex = regex_opt(regex.words, prefix=regex.prefix,
827
+ suffix=regex.suffix)
828
+ else:
829
+ rex = regex
830
+ compiled = re.compile(rex, rflags)
831
+
832
+ def match_func(text, pos, endpos=sys.maxsize):
833
+ info = cls._prof_data[-1].setdefault((state, rex), [0, 0.0])
834
+ t0 = time.time()
835
+ res = compiled.match(text, pos, endpos)
836
+ t1 = time.time()
837
+ info[0] += 1
838
+ info[1] += t1 - t0
839
+ return res
840
+ return match_func
841
+
842
+
843
+ @add_metaclass(ProfilingRegexLexerMeta)
844
+ class ProfilingRegexLexer(RegexLexer):
845
+ """Drop-in replacement for RegexLexer that does profiling of its regexes."""
846
+
847
+ _prof_data = []
848
+ _prof_sort_index = 4 # defaults to time per call
849
+
850
+ def get_tokens_unprocessed(self, text, stack=('root',)):
851
+ # this needs to be a stack, since using(this) will produce nested calls
852
+ self.__class__._prof_data.append({})
853
+ for tok in RegexLexer.get_tokens_unprocessed(self, text, stack):
854
+ yield tok
855
+ rawdata = self.__class__._prof_data.pop()
856
+ data = sorted(((s, repr(r).strip('u\'').replace('\\\\', '\\')[:65],
857
+ n, 1000 * t, 1000 * t / n)
858
+ for ((s, r), (n, t)) in rawdata.items()),
859
+ key=lambda x: x[self._prof_sort_index],
860
+ reverse=True)
861
+ sum_total = sum(x[3] for x in data)
862
+
863
+ print()
864
+ print('Profiling result for %s lexing %d chars in %.3f ms' %
865
+ (self.__class__.__name__, len(text), sum_total))
866
+ print('=' * 110)
867
+ print('%-20s %-64s ncalls tottime percall' % ('state', 'regex'))
868
+ print('-' * 110)
869
+ for d in data:
870
+ print('%-20s %-65s %5d %8.4f %8.4f' % d)
871
+ print('=' * 110)