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.
- package_readme.md +97 -0
- wandb/__init__.py +246 -0
- wandb/__init__.pyi +1197 -0
- wandb/__main__.py +3 -0
- wandb/_globals.py +19 -0
- wandb/agents/__init__.py +0 -0
- wandb/agents/pyagent.py +363 -0
- wandb/analytics/__init__.py +3 -0
- wandb/analytics/sentry.py +263 -0
- wandb/apis/__init__.py +48 -0
- wandb/apis/attrs.py +51 -0
- wandb/apis/importers/__init__.py +1 -0
- wandb/apis/importers/internals/internal.py +385 -0
- wandb/apis/importers/internals/protocols.py +103 -0
- wandb/apis/importers/internals/util.py +78 -0
- wandb/apis/importers/mlflow.py +254 -0
- wandb/apis/importers/validation.py +108 -0
- wandb/apis/importers/wandb.py +1603 -0
- wandb/apis/internal.py +232 -0
- wandb/apis/normalize.py +73 -0
- wandb/apis/paginator.py +81 -0
- wandb/apis/public/__init__.py +34 -0
- wandb/apis/public/api.py +1387 -0
- wandb/apis/public/artifacts.py +1095 -0
- wandb/apis/public/const.py +4 -0
- wandb/apis/public/files.py +263 -0
- wandb/apis/public/history.py +149 -0
- wandb/apis/public/jobs.py +653 -0
- wandb/apis/public/projects.py +154 -0
- wandb/apis/public/query_generator.py +166 -0
- wandb/apis/public/reports.py +458 -0
- wandb/apis/public/runs.py +1012 -0
- wandb/apis/public/sweeps.py +240 -0
- wandb/apis/public/teams.py +198 -0
- wandb/apis/public/users.py +136 -0
- wandb/apis/public/utils.py +68 -0
- wandb/apis/reports/__init__.py +1 -0
- wandb/apis/reports/v1/__init__.py +8 -0
- wandb/apis/reports/v2/__init__.py +8 -0
- wandb/apis/workspaces/__init__.py +8 -0
- wandb/beta/workflows.py +288 -0
- wandb/bin/gpu_stats +0 -0
- wandb/bin/wandb-core +0 -0
- wandb/cli/__init__.py +0 -0
- wandb/cli/beta.py +178 -0
- wandb/cli/cli.py +2812 -0
- wandb/data_types.py +66 -0
- wandb/docker/__init__.py +343 -0
- wandb/docker/auth.py +435 -0
- wandb/docker/wandb-entrypoint.sh +33 -0
- wandb/docker/www_authenticate.py +94 -0
- wandb/env.py +513 -0
- wandb/errors/__init__.py +17 -0
- wandb/errors/errors.py +37 -0
- wandb/errors/links.py +73 -0
- wandb/errors/term.py +415 -0
- wandb/errors/util.py +57 -0
- wandb/errors/warnings.py +2 -0
- wandb/filesync/__init__.py +0 -0
- wandb/filesync/dir_watcher.py +403 -0
- wandb/filesync/stats.py +100 -0
- wandb/filesync/step_checksum.py +142 -0
- wandb/filesync/step_prepare.py +179 -0
- wandb/filesync/step_upload.py +287 -0
- wandb/filesync/upload_job.py +142 -0
- wandb/integration/__init__.py +0 -0
- wandb/integration/catboost/__init__.py +5 -0
- wandb/integration/catboost/catboost.py +178 -0
- wandb/integration/cohere/__init__.py +3 -0
- wandb/integration/cohere/cohere.py +21 -0
- wandb/integration/cohere/resolver.py +347 -0
- wandb/integration/diffusers/__init__.py +3 -0
- wandb/integration/diffusers/autologger.py +76 -0
- wandb/integration/diffusers/pipeline_resolver.py +50 -0
- wandb/integration/diffusers/resolvers/__init__.py +9 -0
- wandb/integration/diffusers/resolvers/multimodal.py +882 -0
- wandb/integration/diffusers/resolvers/utils.py +102 -0
- wandb/integration/fastai/__init__.py +245 -0
- wandb/integration/gym/__init__.py +99 -0
- wandb/integration/huggingface/__init__.py +3 -0
- wandb/integration/huggingface/huggingface.py +18 -0
- wandb/integration/huggingface/resolver.py +213 -0
- wandb/integration/keras/__init__.py +11 -0
- wandb/integration/keras/callbacks/__init__.py +5 -0
- wandb/integration/keras/callbacks/metrics_logger.py +129 -0
- wandb/integration/keras/callbacks/model_checkpoint.py +188 -0
- wandb/integration/keras/callbacks/tables_builder.py +228 -0
- wandb/integration/keras/keras.py +1089 -0
- wandb/integration/kfp/__init__.py +6 -0
- wandb/integration/kfp/helpers.py +28 -0
- wandb/integration/kfp/kfp_patch.py +334 -0
- wandb/integration/kfp/wandb_logging.py +182 -0
- wandb/integration/langchain/__init__.py +3 -0
- wandb/integration/langchain/wandb_tracer.py +48 -0
- wandb/integration/lightgbm/__init__.py +239 -0
- wandb/integration/lightning/__init__.py +0 -0
- wandb/integration/lightning/fabric/__init__.py +3 -0
- wandb/integration/lightning/fabric/logger.py +764 -0
- wandb/integration/metaflow/__init__.py +3 -0
- wandb/integration/metaflow/metaflow.py +383 -0
- wandb/integration/openai/__init__.py +3 -0
- wandb/integration/openai/fine_tuning.py +480 -0
- wandb/integration/openai/openai.py +22 -0
- wandb/integration/openai/resolver.py +240 -0
- wandb/integration/prodigy/__init__.py +3 -0
- wandb/integration/prodigy/prodigy.py +299 -0
- wandb/integration/sacred/__init__.py +117 -0
- wandb/integration/sagemaker/__init__.py +12 -0
- wandb/integration/sagemaker/auth.py +28 -0
- wandb/integration/sagemaker/config.py +49 -0
- wandb/integration/sagemaker/files.py +3 -0
- wandb/integration/sagemaker/resources.py +34 -0
- wandb/integration/sb3/__init__.py +3 -0
- wandb/integration/sb3/sb3.py +147 -0
- wandb/integration/sklearn/__init__.py +37 -0
- wandb/integration/sklearn/calculate/__init__.py +32 -0
- wandb/integration/sklearn/calculate/calibration_curves.py +125 -0
- wandb/integration/sklearn/calculate/class_proportions.py +68 -0
- wandb/integration/sklearn/calculate/confusion_matrix.py +93 -0
- wandb/integration/sklearn/calculate/decision_boundaries.py +40 -0
- wandb/integration/sklearn/calculate/elbow_curve.py +55 -0
- wandb/integration/sklearn/calculate/feature_importances.py +67 -0
- wandb/integration/sklearn/calculate/learning_curve.py +64 -0
- wandb/integration/sklearn/calculate/outlier_candidates.py +69 -0
- wandb/integration/sklearn/calculate/residuals.py +86 -0
- wandb/integration/sklearn/calculate/silhouette.py +118 -0
- wandb/integration/sklearn/calculate/summary_metrics.py +62 -0
- wandb/integration/sklearn/plot/__init__.py +35 -0
- wandb/integration/sklearn/plot/classifier.py +329 -0
- wandb/integration/sklearn/plot/clusterer.py +146 -0
- wandb/integration/sklearn/plot/regressor.py +121 -0
- wandb/integration/sklearn/plot/shared.py +91 -0
- wandb/integration/sklearn/utils.py +183 -0
- wandb/integration/tensorboard/__init__.py +10 -0
- wandb/integration/tensorboard/log.py +354 -0
- wandb/integration/tensorboard/monkeypatch.py +186 -0
- wandb/integration/tensorflow/__init__.py +5 -0
- wandb/integration/tensorflow/estimator_hook.py +54 -0
- wandb/integration/torch/__init__.py +0 -0
- wandb/integration/torch/wandb_torch.py +554 -0
- wandb/integration/ultralytics/__init__.py +11 -0
- wandb/integration/ultralytics/bbox_utils.py +215 -0
- wandb/integration/ultralytics/callback.py +524 -0
- wandb/integration/ultralytics/classification_utils.py +83 -0
- wandb/integration/ultralytics/mask_utils.py +202 -0
- wandb/integration/ultralytics/pose_utils.py +103 -0
- wandb/integration/xgboost/__init__.py +11 -0
- wandb/integration/xgboost/xgboost.py +189 -0
- wandb/integration/yolov8/__init__.py +0 -0
- wandb/integration/yolov8/yolov8.py +284 -0
- wandb/jupyter.py +513 -0
- wandb/mpmain/__init__.py +0 -0
- wandb/mpmain/__main__.py +1 -0
- wandb/old/__init__.py +0 -0
- wandb/old/core.py +53 -0
- wandb/old/settings.py +173 -0
- wandb/old/summary.py +440 -0
- wandb/plot/__init__.py +28 -0
- wandb/plot/bar.py +70 -0
- wandb/plot/confusion_matrix.py +181 -0
- wandb/plot/custom_chart.py +124 -0
- wandb/plot/histogram.py +65 -0
- wandb/plot/line.py +74 -0
- wandb/plot/line_series.py +176 -0
- wandb/plot/pr_curve.py +185 -0
- wandb/plot/roc_curve.py +163 -0
- wandb/plot/scatter.py +66 -0
- wandb/plot/utils.py +183 -0
- wandb/plot/viz.py +41 -0
- wandb/proto/__init__.py +0 -0
- wandb/proto/v3/__init__.py +0 -0
- wandb/proto/v3/wandb_base_pb2.py +55 -0
- wandb/proto/v3/wandb_internal_pb2.py +1658 -0
- wandb/proto/v3/wandb_server_pb2.py +228 -0
- wandb/proto/v3/wandb_settings_pb2.py +122 -0
- wandb/proto/v3/wandb_telemetry_pb2.py +106 -0
- wandb/proto/v4/__init__.py +0 -0
- wandb/proto/v4/wandb_base_pb2.py +30 -0
- wandb/proto/v4/wandb_internal_pb2.py +370 -0
- wandb/proto/v4/wandb_server_pb2.py +67 -0
- wandb/proto/v4/wandb_settings_pb2.py +47 -0
- wandb/proto/v4/wandb_telemetry_pb2.py +41 -0
- wandb/proto/v5/wandb_base_pb2.py +31 -0
- wandb/proto/v5/wandb_internal_pb2.py +371 -0
- wandb/proto/v5/wandb_server_pb2.py +68 -0
- wandb/proto/v5/wandb_settings_pb2.py +48 -0
- wandb/proto/v5/wandb_telemetry_pb2.py +42 -0
- wandb/proto/wandb_base_pb2.py +10 -0
- wandb/proto/wandb_deprecated.py +45 -0
- wandb/proto/wandb_generate_deprecated.py +30 -0
- wandb/proto/wandb_generate_proto.py +49 -0
- wandb/proto/wandb_internal_pb2.py +16 -0
- wandb/proto/wandb_server_pb2.py +10 -0
- wandb/proto/wandb_settings_pb2.py +10 -0
- wandb/proto/wandb_telemetry_pb2.py +10 -0
- wandb/py.typed +0 -0
- wandb/sdk/__init__.py +37 -0
- wandb/sdk/artifacts/__init__.py +0 -0
- wandb/sdk/artifacts/_validators.py +121 -0
- wandb/sdk/artifacts/artifact.py +2364 -0
- wandb/sdk/artifacts/artifact_download_logger.py +43 -0
- wandb/sdk/artifacts/artifact_file_cache.py +249 -0
- wandb/sdk/artifacts/artifact_instance_cache.py +17 -0
- wandb/sdk/artifacts/artifact_manifest.py +75 -0
- wandb/sdk/artifacts/artifact_manifest_entry.py +249 -0
- wandb/sdk/artifacts/artifact_manifests/__init__.py +0 -0
- wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +92 -0
- wandb/sdk/artifacts/artifact_saver.py +265 -0
- wandb/sdk/artifacts/artifact_state.py +11 -0
- wandb/sdk/artifacts/artifact_ttl.py +7 -0
- wandb/sdk/artifacts/exceptions.py +57 -0
- wandb/sdk/artifacts/staging.py +25 -0
- wandb/sdk/artifacts/storage_handler.py +62 -0
- wandb/sdk/artifacts/storage_handlers/__init__.py +0 -0
- wandb/sdk/artifacts/storage_handlers/azure_handler.py +213 -0
- wandb/sdk/artifacts/storage_handlers/gcs_handler.py +224 -0
- wandb/sdk/artifacts/storage_handlers/http_handler.py +114 -0
- wandb/sdk/artifacts/storage_handlers/local_file_handler.py +139 -0
- wandb/sdk/artifacts/storage_handlers/multi_handler.py +56 -0
- wandb/sdk/artifacts/storage_handlers/s3_handler.py +298 -0
- wandb/sdk/artifacts/storage_handlers/tracking_handler.py +72 -0
- wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +135 -0
- wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +74 -0
- wandb/sdk/artifacts/storage_layout.py +6 -0
- wandb/sdk/artifacts/storage_policies/__init__.py +4 -0
- wandb/sdk/artifacts/storage_policies/register.py +1 -0
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +378 -0
- wandb/sdk/artifacts/storage_policy.py +72 -0
- wandb/sdk/backend/__init__.py +0 -0
- wandb/sdk/backend/backend.py +221 -0
- wandb/sdk/data_types/__init__.py +0 -0
- wandb/sdk/data_types/_dtypes.py +918 -0
- wandb/sdk/data_types/_private.py +10 -0
- wandb/sdk/data_types/audio.py +165 -0
- wandb/sdk/data_types/base_types/__init__.py +0 -0
- wandb/sdk/data_types/base_types/json_metadata.py +55 -0
- wandb/sdk/data_types/base_types/media.py +376 -0
- wandb/sdk/data_types/base_types/wb_value.py +282 -0
- wandb/sdk/data_types/bokeh.py +70 -0
- wandb/sdk/data_types/graph.py +405 -0
- wandb/sdk/data_types/helper_types/__init__.py +0 -0
- wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +305 -0
- wandb/sdk/data_types/helper_types/classes.py +159 -0
- wandb/sdk/data_types/helper_types/image_mask.py +241 -0
- wandb/sdk/data_types/histogram.py +94 -0
- wandb/sdk/data_types/html.py +115 -0
- wandb/sdk/data_types/image.py +847 -0
- wandb/sdk/data_types/molecule.py +241 -0
- wandb/sdk/data_types/object_3d.py +470 -0
- wandb/sdk/data_types/plotly.py +82 -0
- wandb/sdk/data_types/saved_model.py +445 -0
- wandb/sdk/data_types/table.py +1204 -0
- wandb/sdk/data_types/trace_tree.py +438 -0
- wandb/sdk/data_types/utils.py +228 -0
- wandb/sdk/data_types/video.py +268 -0
- wandb/sdk/integration_utils/__init__.py +0 -0
- wandb/sdk/integration_utils/auto_logging.py +232 -0
- wandb/sdk/integration_utils/data_logging.py +475 -0
- wandb/sdk/interface/__init__.py +0 -0
- wandb/sdk/interface/constants.py +4 -0
- wandb/sdk/interface/interface.py +1010 -0
- wandb/sdk/interface/interface_queue.py +53 -0
- wandb/sdk/interface/interface_relay.py +53 -0
- wandb/sdk/interface/interface_shared.py +546 -0
- wandb/sdk/interface/interface_sock.py +61 -0
- wandb/sdk/interface/message_future.py +27 -0
- wandb/sdk/interface/message_future_poll.py +50 -0
- wandb/sdk/interface/router.py +115 -0
- wandb/sdk/interface/router_queue.py +41 -0
- wandb/sdk/interface/router_relay.py +37 -0
- wandb/sdk/interface/router_sock.py +36 -0
- wandb/sdk/interface/summary_record.py +67 -0
- wandb/sdk/internal/__init__.py +0 -0
- wandb/sdk/internal/context.py +89 -0
- wandb/sdk/internal/datastore.py +297 -0
- wandb/sdk/internal/file_pusher.py +181 -0
- wandb/sdk/internal/file_stream.py +695 -0
- wandb/sdk/internal/flow_control.py +263 -0
- wandb/sdk/internal/handler.py +905 -0
- wandb/sdk/internal/internal.py +403 -0
- wandb/sdk/internal/internal_api.py +4587 -0
- wandb/sdk/internal/internal_util.py +97 -0
- wandb/sdk/internal/job_builder.py +638 -0
- wandb/sdk/internal/profiler.py +78 -0
- wandb/sdk/internal/progress.py +79 -0
- wandb/sdk/internal/run.py +25 -0
- wandb/sdk/internal/sample.py +70 -0
- wandb/sdk/internal/sender.py +1696 -0
- wandb/sdk/internal/sender_config.py +197 -0
- wandb/sdk/internal/settings_static.py +97 -0
- wandb/sdk/internal/system/__init__.py +0 -0
- wandb/sdk/internal/system/assets/__init__.py +25 -0
- wandb/sdk/internal/system/assets/aggregators.py +37 -0
- wandb/sdk/internal/system/assets/asset_registry.py +20 -0
- wandb/sdk/internal/system/assets/cpu.py +163 -0
- wandb/sdk/internal/system/assets/disk.py +210 -0
- wandb/sdk/internal/system/assets/gpu.py +416 -0
- wandb/sdk/internal/system/assets/gpu_amd.py +233 -0
- wandb/sdk/internal/system/assets/interfaces.py +205 -0
- wandb/sdk/internal/system/assets/ipu.py +177 -0
- wandb/sdk/internal/system/assets/memory.py +166 -0
- wandb/sdk/internal/system/assets/network.py +125 -0
- wandb/sdk/internal/system/assets/open_metrics.py +293 -0
- wandb/sdk/internal/system/assets/tpu.py +154 -0
- wandb/sdk/internal/system/assets/trainium.py +393 -0
- wandb/sdk/internal/system/env_probe_helpers.py +13 -0
- wandb/sdk/internal/system/system_info.py +250 -0
- wandb/sdk/internal/system/system_monitor.py +222 -0
- wandb/sdk/internal/tb_watcher.py +519 -0
- wandb/sdk/internal/thread_local_settings.py +18 -0
- wandb/sdk/internal/writer.py +204 -0
- wandb/sdk/launch/__init__.py +15 -0
- wandb/sdk/launch/_launch.py +331 -0
- wandb/sdk/launch/_launch_add.py +255 -0
- wandb/sdk/launch/_project_spec.py +566 -0
- wandb/sdk/launch/agent/__init__.py +5 -0
- wandb/sdk/launch/agent/agent.py +924 -0
- wandb/sdk/launch/agent/config.py +296 -0
- wandb/sdk/launch/agent/job_status_tracker.py +53 -0
- wandb/sdk/launch/agent/run_queue_item_file_saver.py +39 -0
- wandb/sdk/launch/builder/__init__.py +0 -0
- wandb/sdk/launch/builder/abstract.py +156 -0
- wandb/sdk/launch/builder/build.py +297 -0
- wandb/sdk/launch/builder/context_manager.py +235 -0
- wandb/sdk/launch/builder/docker_builder.py +177 -0
- wandb/sdk/launch/builder/kaniko_builder.py +594 -0
- wandb/sdk/launch/builder/noop.py +58 -0
- wandb/sdk/launch/builder/templates/_wandb_bootstrap.py +188 -0
- wandb/sdk/launch/builder/templates/dockerfile.py +92 -0
- wandb/sdk/launch/create_job.py +528 -0
- wandb/sdk/launch/environment/abstract.py +29 -0
- wandb/sdk/launch/environment/aws_environment.py +322 -0
- wandb/sdk/launch/environment/azure_environment.py +105 -0
- wandb/sdk/launch/environment/gcp_environment.py +335 -0
- wandb/sdk/launch/environment/local_environment.py +65 -0
- wandb/sdk/launch/errors.py +13 -0
- wandb/sdk/launch/git_reference.py +109 -0
- wandb/sdk/launch/inputs/files.py +148 -0
- wandb/sdk/launch/inputs/internal.py +315 -0
- wandb/sdk/launch/inputs/manage.py +113 -0
- wandb/sdk/launch/inputs/schema.py +39 -0
- wandb/sdk/launch/loader.py +249 -0
- wandb/sdk/launch/registry/abstract.py +48 -0
- wandb/sdk/launch/registry/anon.py +29 -0
- wandb/sdk/launch/registry/azure_container_registry.py +124 -0
- wandb/sdk/launch/registry/elastic_container_registry.py +192 -0
- wandb/sdk/launch/registry/google_artifact_registry.py +219 -0
- wandb/sdk/launch/registry/local_registry.py +65 -0
- wandb/sdk/launch/runner/__init__.py +0 -0
- wandb/sdk/launch/runner/abstract.py +185 -0
- wandb/sdk/launch/runner/kubernetes_monitor.py +472 -0
- wandb/sdk/launch/runner/kubernetes_runner.py +963 -0
- wandb/sdk/launch/runner/local_container.py +301 -0
- wandb/sdk/launch/runner/local_process.py +78 -0
- wandb/sdk/launch/runner/sagemaker_runner.py +426 -0
- wandb/sdk/launch/runner/vertex_runner.py +230 -0
- wandb/sdk/launch/sweeps/__init__.py +37 -0
- wandb/sdk/launch/sweeps/scheduler.py +740 -0
- wandb/sdk/launch/sweeps/scheduler_sweep.py +90 -0
- wandb/sdk/launch/sweeps/utils.py +316 -0
- wandb/sdk/launch/utils.py +747 -0
- wandb/sdk/launch/wandb_reference.py +138 -0
- wandb/sdk/lib/__init__.py +5 -0
- wandb/sdk/lib/apikey.py +269 -0
- wandb/sdk/lib/capped_dict.py +26 -0
- wandb/sdk/lib/config_util.py +101 -0
- wandb/sdk/lib/credentials.py +141 -0
- wandb/sdk/lib/deprecate.py +42 -0
- wandb/sdk/lib/disabled.py +29 -0
- wandb/sdk/lib/exit_hooks.py +54 -0
- wandb/sdk/lib/file_stream_utils.py +118 -0
- wandb/sdk/lib/filenames.py +64 -0
- wandb/sdk/lib/filesystem.py +372 -0
- wandb/sdk/lib/fsm.py +180 -0
- wandb/sdk/lib/gitlib.py +239 -0
- wandb/sdk/lib/gql_request.py +65 -0
- wandb/sdk/lib/handler_util.py +21 -0
- wandb/sdk/lib/hashutil.py +84 -0
- wandb/sdk/lib/import_hooks.py +275 -0
- wandb/sdk/lib/ipython.py +126 -0
- wandb/sdk/lib/json_util.py +80 -0
- wandb/sdk/lib/lazyloader.py +63 -0
- wandb/sdk/lib/mailbox.py +456 -0
- wandb/sdk/lib/module.py +78 -0
- wandb/sdk/lib/paths.py +106 -0
- wandb/sdk/lib/preinit.py +42 -0
- wandb/sdk/lib/printer.py +548 -0
- wandb/sdk/lib/progress.py +279 -0
- wandb/sdk/lib/proto_util.py +90 -0
- wandb/sdk/lib/redirect.py +845 -0
- wandb/sdk/lib/retry.py +289 -0
- wandb/sdk/lib/run_moment.py +72 -0
- wandb/sdk/lib/runid.py +12 -0
- wandb/sdk/lib/server.py +38 -0
- wandb/sdk/lib/service_connection.py +216 -0
- wandb/sdk/lib/service_token.py +94 -0
- wandb/sdk/lib/sock_client.py +290 -0
- wandb/sdk/lib/sparkline.py +44 -0
- wandb/sdk/lib/telemetry.py +100 -0
- wandb/sdk/lib/timed_input.py +133 -0
- wandb/sdk/lib/timer.py +19 -0
- wandb/sdk/service/__init__.py +0 -0
- wandb/sdk/service/_startup_debug.py +22 -0
- wandb/sdk/service/port_file.py +53 -0
- wandb/sdk/service/server.py +107 -0
- wandb/sdk/service/server_sock.py +274 -0
- wandb/sdk/service/service.py +242 -0
- wandb/sdk/service/streams.py +425 -0
- wandb/sdk/verify/__init__.py +0 -0
- wandb/sdk/verify/verify.py +501 -0
- wandb/sdk/wandb_alerts.py +12 -0
- wandb/sdk/wandb_config.py +322 -0
- wandb/sdk/wandb_helper.py +54 -0
- wandb/sdk/wandb_init.py +1313 -0
- wandb/sdk/wandb_login.py +339 -0
- wandb/sdk/wandb_metric.py +110 -0
- wandb/sdk/wandb_require.py +94 -0
- wandb/sdk/wandb_require_helpers.py +44 -0
- wandb/sdk/wandb_run.py +4066 -0
- wandb/sdk/wandb_settings.py +1309 -0
- wandb/sdk/wandb_setup.py +402 -0
- wandb/sdk/wandb_summary.py +150 -0
- wandb/sdk/wandb_sweep.py +119 -0
- wandb/sdk/wandb_sync.py +82 -0
- wandb/sdk/wandb_watch.py +150 -0
- wandb/sklearn.py +35 -0
- wandb/sync/__init__.py +3 -0
- wandb/sync/sync.py +442 -0
- wandb/trigger.py +29 -0
- wandb/util.py +1955 -0
- wandb/vendor/__init__.py +0 -0
- wandb/vendor/gql-0.2.0/setup.py +40 -0
- wandb/vendor/gql-0.2.0/tests/__init__.py +0 -0
- wandb/vendor/gql-0.2.0/tests/starwars/__init__.py +0 -0
- wandb/vendor/gql-0.2.0/tests/starwars/fixtures.py +96 -0
- wandb/vendor/gql-0.2.0/tests/starwars/schema.py +146 -0
- wandb/vendor/gql-0.2.0/tests/starwars/test_dsl.py +293 -0
- wandb/vendor/gql-0.2.0/tests/starwars/test_query.py +355 -0
- wandb/vendor/gql-0.2.0/tests/starwars/test_validation.py +171 -0
- wandb/vendor/gql-0.2.0/tests/test_client.py +31 -0
- wandb/vendor/gql-0.2.0/tests/test_transport.py +89 -0
- wandb/vendor/gql-0.2.0/wandb_gql/__init__.py +4 -0
- wandb/vendor/gql-0.2.0/wandb_gql/client.py +75 -0
- wandb/vendor/gql-0.2.0/wandb_gql/dsl.py +152 -0
- wandb/vendor/gql-0.2.0/wandb_gql/gql.py +10 -0
- wandb/vendor/gql-0.2.0/wandb_gql/transport/__init__.py +0 -0
- wandb/vendor/gql-0.2.0/wandb_gql/transport/http.py +6 -0
- wandb/vendor/gql-0.2.0/wandb_gql/transport/local_schema.py +15 -0
- wandb/vendor/gql-0.2.0/wandb_gql/transport/requests.py +46 -0
- wandb/vendor/gql-0.2.0/wandb_gql/utils.py +21 -0
- wandb/vendor/graphql-core-1.1/setup.py +86 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/__init__.py +287 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/error/__init__.py +6 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/error/base.py +42 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/error/format_error.py +11 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/error/located_error.py +29 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/error/syntax_error.py +36 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/__init__.py +26 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/base.py +311 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executor.py +398 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/__init__.py +0 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/asyncio.py +53 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/gevent.py +22 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/process.py +32 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/sync.py +7 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/thread.py +35 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/executors/utils.py +6 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/__init__.py +0 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/executor.py +66 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/fragment.py +252 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/resolver.py +151 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/experimental/utils.py +7 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/middleware.py +57 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/execution/values.py +145 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/graphql.py +60 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/__init__.py +0 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/ast.py +1349 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/base.py +19 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/lexer.py +435 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/location.py +30 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/parser.py +779 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/printer.py +193 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/source.py +18 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/visitor.py +222 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/language/visitor_meta.py +82 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/__init__.py +0 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/cached_property.py +17 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/contain_subset.py +28 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/default_ordered_dict.py +40 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/ordereddict.py +8 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/pair_set.py +43 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/pyutils/version.py +78 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/__init__.py +67 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/definition.py +619 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/directives.py +132 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/introspection.py +440 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/scalars.py +131 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/schema.py +100 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/type/typemap.py +145 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/__init__.py +0 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/assert_valid_name.py +9 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_from_value.py +65 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_to_code.py +49 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/ast_to_dict.py +24 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/base.py +75 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/build_ast_schema.py +291 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/build_client_schema.py +250 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/concat_ast.py +9 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/extend_schema.py +357 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/get_field_def.py +27 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/get_operation_ast.py +21 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/introspection_query.py +90 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/is_valid_literal_value.py +67 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/is_valid_value.py +66 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/quoted_or_list.py +21 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/schema_printer.py +168 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/suggestion_list.py +56 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_comparators.py +69 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_from_ast.py +21 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/type_info.py +149 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/utils/value_from_ast.py +69 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/__init__.py +4 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/__init__.py +79 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/arguments_of_correct_type.py +24 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/base.py +8 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/default_values_of_correct_type.py +44 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/fields_on_correct_type.py +113 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/fragments_on_composite_types.py +33 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_argument_names.py +70 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_directives.py +97 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_fragment_names.py +19 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/known_type_names.py +43 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/lone_anonymous_operation.py +23 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_fragment_cycles.py +59 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_undefined_variables.py +36 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_unused_fragments.py +38 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/no_unused_variables.py +37 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/overlapping_fields_can_be_merged.py +529 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/possible_fragment_spreads.py +44 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/provided_non_null_arguments.py +46 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/scalar_leafs.py +33 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_argument_names.py +32 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_fragment_names.py +28 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_input_field_names.py +33 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_operation_names.py +31 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/unique_variable_names.py +27 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/variables_are_input_types.py +21 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/rules/variables_in_allowed_position.py +53 -0
- wandb/vendor/graphql-core-1.1/wandb_graphql/validation/validation.py +158 -0
- wandb/vendor/promise-2.3.0/conftest.py +30 -0
- wandb/vendor/promise-2.3.0/setup.py +64 -0
- wandb/vendor/promise-2.3.0/tests/__init__.py +0 -0
- wandb/vendor/promise-2.3.0/tests/conftest.py +8 -0
- wandb/vendor/promise-2.3.0/tests/test_awaitable.py +32 -0
- wandb/vendor/promise-2.3.0/tests/test_awaitable_35.py +47 -0
- wandb/vendor/promise-2.3.0/tests/test_benchmark.py +116 -0
- wandb/vendor/promise-2.3.0/tests/test_complex_threads.py +23 -0
- wandb/vendor/promise-2.3.0/tests/test_dataloader.py +452 -0
- wandb/vendor/promise-2.3.0/tests/test_dataloader_awaitable_35.py +99 -0
- wandb/vendor/promise-2.3.0/tests/test_dataloader_extra.py +65 -0
- wandb/vendor/promise-2.3.0/tests/test_extra.py +670 -0
- wandb/vendor/promise-2.3.0/tests/test_issues.py +132 -0
- wandb/vendor/promise-2.3.0/tests/test_promise_list.py +70 -0
- wandb/vendor/promise-2.3.0/tests/test_spec.py +584 -0
- wandb/vendor/promise-2.3.0/tests/test_thread_safety.py +115 -0
- wandb/vendor/promise-2.3.0/tests/utils.py +3 -0
- wandb/vendor/promise-2.3.0/wandb_promise/__init__.py +38 -0
- wandb/vendor/promise-2.3.0/wandb_promise/async_.py +135 -0
- wandb/vendor/promise-2.3.0/wandb_promise/compat.py +32 -0
- wandb/vendor/promise-2.3.0/wandb_promise/dataloader.py +326 -0
- wandb/vendor/promise-2.3.0/wandb_promise/iterate_promise.py +12 -0
- wandb/vendor/promise-2.3.0/wandb_promise/promise.py +848 -0
- wandb/vendor/promise-2.3.0/wandb_promise/promise_list.py +151 -0
- wandb/vendor/promise-2.3.0/wandb_promise/pyutils/__init__.py +0 -0
- wandb/vendor/promise-2.3.0/wandb_promise/pyutils/version.py +83 -0
- wandb/vendor/promise-2.3.0/wandb_promise/schedulers/__init__.py +0 -0
- wandb/vendor/promise-2.3.0/wandb_promise/schedulers/asyncio.py +22 -0
- wandb/vendor/promise-2.3.0/wandb_promise/schedulers/gevent.py +21 -0
- wandb/vendor/promise-2.3.0/wandb_promise/schedulers/immediate.py +27 -0
- wandb/vendor/promise-2.3.0/wandb_promise/schedulers/thread.py +18 -0
- wandb/vendor/promise-2.3.0/wandb_promise/utils.py +56 -0
- wandb/vendor/pygments/__init__.py +90 -0
- wandb/vendor/pygments/cmdline.py +568 -0
- wandb/vendor/pygments/console.py +74 -0
- wandb/vendor/pygments/filter.py +74 -0
- wandb/vendor/pygments/filters/__init__.py +350 -0
- wandb/vendor/pygments/formatter.py +95 -0
- wandb/vendor/pygments/formatters/__init__.py +153 -0
- wandb/vendor/pygments/formatters/_mapping.py +85 -0
- wandb/vendor/pygments/formatters/bbcode.py +109 -0
- wandb/vendor/pygments/formatters/html.py +851 -0
- wandb/vendor/pygments/formatters/img.py +600 -0
- wandb/vendor/pygments/formatters/irc.py +182 -0
- wandb/vendor/pygments/formatters/latex.py +482 -0
- wandb/vendor/pygments/formatters/other.py +160 -0
- wandb/vendor/pygments/formatters/rtf.py +147 -0
- wandb/vendor/pygments/formatters/svg.py +153 -0
- wandb/vendor/pygments/formatters/terminal.py +136 -0
- wandb/vendor/pygments/formatters/terminal256.py +309 -0
- wandb/vendor/pygments/lexer.py +871 -0
- wandb/vendor/pygments/lexers/__init__.py +329 -0
- wandb/vendor/pygments/lexers/_asy_builtins.py +1645 -0
- wandb/vendor/pygments/lexers/_cl_builtins.py +232 -0
- wandb/vendor/pygments/lexers/_cocoa_builtins.py +72 -0
- wandb/vendor/pygments/lexers/_csound_builtins.py +1346 -0
- wandb/vendor/pygments/lexers/_lasso_builtins.py +5327 -0
- wandb/vendor/pygments/lexers/_lua_builtins.py +295 -0
- wandb/vendor/pygments/lexers/_mapping.py +500 -0
- wandb/vendor/pygments/lexers/_mql_builtins.py +1172 -0
- wandb/vendor/pygments/lexers/_openedge_builtins.py +2547 -0
- wandb/vendor/pygments/lexers/_php_builtins.py +4756 -0
- wandb/vendor/pygments/lexers/_postgres_builtins.py +621 -0
- wandb/vendor/pygments/lexers/_scilab_builtins.py +3094 -0
- wandb/vendor/pygments/lexers/_sourcemod_builtins.py +1163 -0
- wandb/vendor/pygments/lexers/_stan_builtins.py +532 -0
- wandb/vendor/pygments/lexers/_stata_builtins.py +419 -0
- wandb/vendor/pygments/lexers/_tsql_builtins.py +1004 -0
- wandb/vendor/pygments/lexers/_vim_builtins.py +1939 -0
- wandb/vendor/pygments/lexers/actionscript.py +240 -0
- wandb/vendor/pygments/lexers/agile.py +24 -0
- wandb/vendor/pygments/lexers/algebra.py +221 -0
- wandb/vendor/pygments/lexers/ambient.py +76 -0
- wandb/vendor/pygments/lexers/ampl.py +87 -0
- wandb/vendor/pygments/lexers/apl.py +101 -0
- wandb/vendor/pygments/lexers/archetype.py +318 -0
- wandb/vendor/pygments/lexers/asm.py +641 -0
- wandb/vendor/pygments/lexers/automation.py +374 -0
- wandb/vendor/pygments/lexers/basic.py +500 -0
- wandb/vendor/pygments/lexers/bibtex.py +160 -0
- wandb/vendor/pygments/lexers/business.py +612 -0
- wandb/vendor/pygments/lexers/c_cpp.py +252 -0
- wandb/vendor/pygments/lexers/c_like.py +541 -0
- wandb/vendor/pygments/lexers/capnproto.py +78 -0
- wandb/vendor/pygments/lexers/chapel.py +102 -0
- wandb/vendor/pygments/lexers/clean.py +288 -0
- wandb/vendor/pygments/lexers/compiled.py +34 -0
- wandb/vendor/pygments/lexers/configs.py +833 -0
- wandb/vendor/pygments/lexers/console.py +114 -0
- wandb/vendor/pygments/lexers/crystal.py +393 -0
- wandb/vendor/pygments/lexers/csound.py +366 -0
- wandb/vendor/pygments/lexers/css.py +689 -0
- wandb/vendor/pygments/lexers/d.py +251 -0
- wandb/vendor/pygments/lexers/dalvik.py +125 -0
- wandb/vendor/pygments/lexers/data.py +555 -0
- wandb/vendor/pygments/lexers/diff.py +165 -0
- wandb/vendor/pygments/lexers/dotnet.py +691 -0
- wandb/vendor/pygments/lexers/dsls.py +878 -0
- wandb/vendor/pygments/lexers/dylan.py +289 -0
- wandb/vendor/pygments/lexers/ecl.py +125 -0
- wandb/vendor/pygments/lexers/eiffel.py +65 -0
- wandb/vendor/pygments/lexers/elm.py +121 -0
- wandb/vendor/pygments/lexers/erlang.py +533 -0
- wandb/vendor/pygments/lexers/esoteric.py +277 -0
- wandb/vendor/pygments/lexers/ezhil.py +69 -0
- wandb/vendor/pygments/lexers/factor.py +344 -0
- wandb/vendor/pygments/lexers/fantom.py +250 -0
- wandb/vendor/pygments/lexers/felix.py +273 -0
- wandb/vendor/pygments/lexers/forth.py +177 -0
- wandb/vendor/pygments/lexers/fortran.py +205 -0
- wandb/vendor/pygments/lexers/foxpro.py +428 -0
- wandb/vendor/pygments/lexers/functional.py +21 -0
- wandb/vendor/pygments/lexers/go.py +101 -0
- wandb/vendor/pygments/lexers/grammar_notation.py +213 -0
- wandb/vendor/pygments/lexers/graph.py +80 -0
- wandb/vendor/pygments/lexers/graphics.py +553 -0
- wandb/vendor/pygments/lexers/haskell.py +843 -0
- wandb/vendor/pygments/lexers/haxe.py +936 -0
- wandb/vendor/pygments/lexers/hdl.py +382 -0
- wandb/vendor/pygments/lexers/hexdump.py +103 -0
- wandb/vendor/pygments/lexers/html.py +602 -0
- wandb/vendor/pygments/lexers/idl.py +270 -0
- wandb/vendor/pygments/lexers/igor.py +288 -0
- wandb/vendor/pygments/lexers/inferno.py +96 -0
- wandb/vendor/pygments/lexers/installers.py +322 -0
- wandb/vendor/pygments/lexers/int_fiction.py +1343 -0
- wandb/vendor/pygments/lexers/iolang.py +63 -0
- wandb/vendor/pygments/lexers/j.py +146 -0
- wandb/vendor/pygments/lexers/javascript.py +1525 -0
- wandb/vendor/pygments/lexers/julia.py +333 -0
- wandb/vendor/pygments/lexers/jvm.py +1573 -0
- wandb/vendor/pygments/lexers/lisp.py +2621 -0
- wandb/vendor/pygments/lexers/make.py +202 -0
- wandb/vendor/pygments/lexers/markup.py +595 -0
- wandb/vendor/pygments/lexers/math.py +21 -0
- wandb/vendor/pygments/lexers/matlab.py +663 -0
- wandb/vendor/pygments/lexers/ml.py +769 -0
- wandb/vendor/pygments/lexers/modeling.py +358 -0
- wandb/vendor/pygments/lexers/modula2.py +1561 -0
- wandb/vendor/pygments/lexers/monte.py +204 -0
- wandb/vendor/pygments/lexers/ncl.py +894 -0
- wandb/vendor/pygments/lexers/nimrod.py +159 -0
- wandb/vendor/pygments/lexers/nit.py +64 -0
- wandb/vendor/pygments/lexers/nix.py +136 -0
- wandb/vendor/pygments/lexers/oberon.py +105 -0
- wandb/vendor/pygments/lexers/objective.py +504 -0
- wandb/vendor/pygments/lexers/ooc.py +85 -0
- wandb/vendor/pygments/lexers/other.py +41 -0
- wandb/vendor/pygments/lexers/parasail.py +79 -0
- wandb/vendor/pygments/lexers/parsers.py +835 -0
- wandb/vendor/pygments/lexers/pascal.py +644 -0
- wandb/vendor/pygments/lexers/pawn.py +199 -0
- wandb/vendor/pygments/lexers/perl.py +620 -0
- wandb/vendor/pygments/lexers/php.py +267 -0
- wandb/vendor/pygments/lexers/praat.py +294 -0
- wandb/vendor/pygments/lexers/prolog.py +306 -0
- wandb/vendor/pygments/lexers/python.py +939 -0
- wandb/vendor/pygments/lexers/qvt.py +152 -0
- wandb/vendor/pygments/lexers/r.py +453 -0
- wandb/vendor/pygments/lexers/rdf.py +270 -0
- wandb/vendor/pygments/lexers/rebol.py +431 -0
- wandb/vendor/pygments/lexers/resource.py +85 -0
- wandb/vendor/pygments/lexers/rnc.py +67 -0
- wandb/vendor/pygments/lexers/roboconf.py +82 -0
- wandb/vendor/pygments/lexers/robotframework.py +560 -0
- wandb/vendor/pygments/lexers/ruby.py +519 -0
- wandb/vendor/pygments/lexers/rust.py +220 -0
- wandb/vendor/pygments/lexers/sas.py +228 -0
- wandb/vendor/pygments/lexers/scripting.py +1222 -0
- wandb/vendor/pygments/lexers/shell.py +794 -0
- wandb/vendor/pygments/lexers/smalltalk.py +195 -0
- wandb/vendor/pygments/lexers/smv.py +79 -0
- wandb/vendor/pygments/lexers/snobol.py +83 -0
- wandb/vendor/pygments/lexers/special.py +103 -0
- wandb/vendor/pygments/lexers/sql.py +681 -0
- wandb/vendor/pygments/lexers/stata.py +108 -0
- wandb/vendor/pygments/lexers/supercollider.py +90 -0
- wandb/vendor/pygments/lexers/tcl.py +145 -0
- wandb/vendor/pygments/lexers/templates.py +2283 -0
- wandb/vendor/pygments/lexers/testing.py +207 -0
- wandb/vendor/pygments/lexers/text.py +25 -0
- wandb/vendor/pygments/lexers/textedit.py +169 -0
- wandb/vendor/pygments/lexers/textfmts.py +297 -0
- wandb/vendor/pygments/lexers/theorem.py +458 -0
- wandb/vendor/pygments/lexers/trafficscript.py +54 -0
- wandb/vendor/pygments/lexers/typoscript.py +226 -0
- wandb/vendor/pygments/lexers/urbi.py +133 -0
- wandb/vendor/pygments/lexers/varnish.py +190 -0
- wandb/vendor/pygments/lexers/verification.py +111 -0
- wandb/vendor/pygments/lexers/web.py +24 -0
- wandb/vendor/pygments/lexers/webmisc.py +988 -0
- wandb/vendor/pygments/lexers/whiley.py +116 -0
- wandb/vendor/pygments/lexers/x10.py +69 -0
- wandb/vendor/pygments/modeline.py +44 -0
- wandb/vendor/pygments/plugin.py +68 -0
- wandb/vendor/pygments/regexopt.py +92 -0
- wandb/vendor/pygments/scanner.py +105 -0
- wandb/vendor/pygments/sphinxext.py +158 -0
- wandb/vendor/pygments/style.py +155 -0
- wandb/vendor/pygments/styles/__init__.py +80 -0
- wandb/vendor/pygments/styles/abap.py +29 -0
- wandb/vendor/pygments/styles/algol.py +63 -0
- wandb/vendor/pygments/styles/algol_nu.py +63 -0
- wandb/vendor/pygments/styles/arduino.py +98 -0
- wandb/vendor/pygments/styles/autumn.py +65 -0
- wandb/vendor/pygments/styles/borland.py +51 -0
- wandb/vendor/pygments/styles/bw.py +49 -0
- wandb/vendor/pygments/styles/colorful.py +81 -0
- wandb/vendor/pygments/styles/default.py +73 -0
- wandb/vendor/pygments/styles/emacs.py +72 -0
- wandb/vendor/pygments/styles/friendly.py +72 -0
- wandb/vendor/pygments/styles/fruity.py +42 -0
- wandb/vendor/pygments/styles/igor.py +29 -0
- wandb/vendor/pygments/styles/lovelace.py +97 -0
- wandb/vendor/pygments/styles/manni.py +75 -0
- wandb/vendor/pygments/styles/monokai.py +106 -0
- wandb/vendor/pygments/styles/murphy.py +80 -0
- wandb/vendor/pygments/styles/native.py +65 -0
- wandb/vendor/pygments/styles/paraiso_dark.py +125 -0
- wandb/vendor/pygments/styles/paraiso_light.py +125 -0
- wandb/vendor/pygments/styles/pastie.py +75 -0
- wandb/vendor/pygments/styles/perldoc.py +69 -0
- wandb/vendor/pygments/styles/rainbow_dash.py +89 -0
- wandb/vendor/pygments/styles/rrt.py +33 -0
- wandb/vendor/pygments/styles/sas.py +44 -0
- wandb/vendor/pygments/styles/stata.py +40 -0
- wandb/vendor/pygments/styles/tango.py +141 -0
- wandb/vendor/pygments/styles/trac.py +63 -0
- wandb/vendor/pygments/styles/vim.py +63 -0
- wandb/vendor/pygments/styles/vs.py +38 -0
- wandb/vendor/pygments/styles/xcode.py +51 -0
- wandb/vendor/pygments/token.py +213 -0
- wandb/vendor/pygments/unistring.py +217 -0
- wandb/vendor/pygments/util.py +388 -0
- wandb/vendor/pynvml/__init__.py +0 -0
- wandb/vendor/pynvml/pynvml.py +4779 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/__init__.py +17 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/events.py +615 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/__init__.py +98 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/api.py +369 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/fsevents.py +172 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/fsevents2.py +239 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify.py +218 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_buffer.py +81 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_c.py +575 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/kqueue.py +730 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/polling.py +145 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/read_directory_changes.py +133 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/winapi.py +348 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/patterns.py +265 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/tricks/__init__.py +174 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/__init__.py +151 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/bricks.py +249 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/compat.py +29 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/decorators.py +198 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/delayed_queue.py +88 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/dirsnapshot.py +293 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/echo.py +157 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/event_backport.py +41 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py +40 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py +57 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py +64 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py +123 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/version.py +28 -0
- wandb/vendor/watchdog_0_9_0/wandb_watchdog/watchmedo.py +577 -0
- wandb/wandb_agent.py +588 -0
- wandb/wandb_controller.py +719 -0
- wandb/wandb_run.py +9 -0
- wandb-0.19.1.dist-info/METADATA +223 -0
- wandb-0.19.1.dist-info/RECORD +822 -0
- wandb-0.19.1.dist-info/WHEEL +5 -0
- wandb-0.19.1.dist-info/entry_points.txt +3 -0
- 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)
|