coiled 1.135.1.dev11__tar.gz → 1.135.2.dev1__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/PKG-INFO +1 -1
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/context.py +1 -19
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/.gitignore +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/LICENSE +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/README.md +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/__main__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/analytics.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/auth.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/batch.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/capture_environment.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/list.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/logs.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/run.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/status.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/util.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/batch/wait.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/azure_logs.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/better_logs.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/crud.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/get_address.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/list.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/logs.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/metrics.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/ssh.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/cluster/utils.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/config.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/core.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/curl.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/diagnostics.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/env.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/file.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/examples/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/examples/exit.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/examples/hello_world.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/examples/nyc_parquet.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/examples/pytorch.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/examples/xarray_nwm.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/hello.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/scripts/fill_ipython.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/scripts/nyc_parquet.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/scripts/pytorch.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/scripts/xarray_nwm.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/hello/utils.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/login.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/mpi.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/notebook/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/notebook/notebook.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/package_sync.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/prefect.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/prefect_serve.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/run.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/amp.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/aws.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/azure.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/entry.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/gcp.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/prometheus.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/setup/util.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/sync.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cli/utils.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/cluster.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/coiled.yaml +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/compatibility.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/config.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/core.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/credentials/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/credentials/aws.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/credentials/google.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/errors.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/exceptions.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/extensions/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/extensions/prefect/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/extensions/prefect/runners.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/extensions/prefect/workers.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/filestore.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/function.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/plugins.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/prefect.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/pypi_conda_map.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/scan.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/software.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/software_utils.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/spans.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/spark.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/types.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/utils.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/cluster.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/cluster_comms.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/core.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/cwi_log_link.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/states.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/widgets/__init__.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/widgets/interface.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/widgets/rich.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/v2/widgets/util.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/coiled/websockets.py +0 -0
- {coiled-1.135.1.dev11 → coiled-1.135.2.dev1}/pyproject.toml +0 -0
|
@@ -8,20 +8,11 @@ from contextlib import contextmanager, nullcontext
|
|
|
8
8
|
from contextvars import ContextVar
|
|
9
9
|
from datetime import datetime, timedelta, timezone
|
|
10
10
|
from logging import getLogger
|
|
11
|
-
from os import environ
|
|
12
11
|
from typing import Any, Awaitable, Callable, Dict, TypeVar, Union, cast
|
|
13
12
|
from urllib.parse import urlencode
|
|
14
13
|
|
|
15
14
|
import aiohttp
|
|
16
15
|
|
|
17
|
-
DDTRACE = environ.get("DD_TRACE_ENABLED", "true").lower() in ["true", "1"]
|
|
18
|
-
if DDTRACE:
|
|
19
|
-
try:
|
|
20
|
-
from ddtrace import tracer
|
|
21
|
-
from ddtrace.propagation.http import HTTPPropagator
|
|
22
|
-
except ImportError:
|
|
23
|
-
DDTRACE = False
|
|
24
|
-
|
|
25
16
|
logger = getLogger(__name__)
|
|
26
17
|
|
|
27
18
|
|
|
@@ -94,12 +85,7 @@ AsyncFuncType = Callable[..., Awaitable[Any]]
|
|
|
94
85
|
|
|
95
86
|
|
|
96
87
|
def get_trace_context(func: Union[SyncFuncType, AsyncFuncType]):
|
|
97
|
-
|
|
98
|
-
return tracer.trace(name=f"{func.__module__}.{func.__qualname__}"), operation_context( # type: ignore
|
|
99
|
-
name=f"{func.__module__}.{func.__qualname__}"
|
|
100
|
-
)
|
|
101
|
-
else:
|
|
102
|
-
return nullcontext(), nullcontext()
|
|
88
|
+
return nullcontext(), nullcontext()
|
|
103
89
|
|
|
104
90
|
|
|
105
91
|
def track_context(func: F) -> F:
|
|
@@ -132,10 +118,6 @@ def create_trace_data() -> Dict[str, str]:
|
|
|
132
118
|
"coiled-session-id": COILED_SESSION_ID,
|
|
133
119
|
"coiled-request-id": random_str(),
|
|
134
120
|
}
|
|
135
|
-
if DDTRACE:
|
|
136
|
-
span = tracer.current_span() # type: ignore
|
|
137
|
-
if span:
|
|
138
|
-
HTTPPropagator.inject(span_context=span.context, headers=trace_data) # type: ignore
|
|
139
121
|
op_id = COILED_OP_CONTEXT.get()
|
|
140
122
|
if op_id:
|
|
141
123
|
trace_data["coiled-operation-id"] = op_id
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|