SYMBOL INDEX (3199 symbols across 290 files) FILE: cmd/cog/cog.go function main (line 8) | func main() { FILE: crates/coglet-python/build.rs function main (line 7) | fn main() { function semver_to_pep440 (line 73) | fn semver_to_pep440(version: &str) -> String { function test_stable_version (line 121) | fn test_stable_version() { function test_alpha (line 127) | fn test_alpha() { function test_beta (line 134) | fn test_beta() { function test_rc (line 140) | fn test_rc() { function test_dev (line 146) | fn test_dev() { FILE: crates/coglet-python/coglet/_impl.pyi class BuildInfo (line 18) | class BuildInfo: method version (line 23) | def version(self) -> builtins.str: ... method git_sha (line 25) | def git_sha(self) -> builtins.str: ... method build_time (line 27) | def build_time(self) -> builtins.str: ... method rustc_version (line 29) | def rustc_version(self) -> builtins.str: ... method __repr__ (line 30) | def __repr__(self) -> builtins.str: ... class CancelationException (line 32) | class CancelationException(builtins.BaseException): class Server (line 44) | class Server: method active (line 54) | def active(self) -> builtins.bool: method serve (line 58) | def serve(self, predictor_ref: typing.Optional[builtins.str] = None, h... method _run_worker (line 62) | def _run_worker(self) -> None: method __repr__ (line 69) | def __repr__(self) -> builtins.str: ... FILE: crates/coglet-python/coglet/_sdk/__init__.pyi class MetricRecorder (line 14) | class MetricRecorder: method record (line 24) | def record( method delete (line 41) | def delete(self, key: builtins.str) -> None: method __setitem__ (line 45) | def __setitem__(self, key: builtins.str, value: typing.Any) -> None: method __delitem__ (line 49) | def __delitem__(self, key: builtins.str) -> None: method __repr__ (line 53) | def __repr__(self) -> builtins.str: ... class Scope (line 56) | class Scope: method metrics (line 64) | def metrics(self) -> MetricRecorder: method context (line 69) | def context(self) -> dict[builtins.str, builtins.str]: method record_metric (line 75) | def record_metric( method __repr__ (line 86) | def __repr__(self) -> builtins.str: ... class _SlotLogWriter (line 89) | class _SlotLogWriter: method closed (line 105) | def closed(self) -> builtins.bool: method encoding (line 110) | def encoding(self) -> typing.Optional[builtins.str]: method newlines (line 115) | def newlines(self) -> typing.Optional[builtins.str]: method buffer (line 120) | def buffer(self) -> typing.Any: method write (line 124) | def write(self, data: builtins.str) -> builtins.int: method emit_data (line 137) | def emit_data(self, data: builtins.str) -> None: method flush (line 141) | def flush(self) -> None: method readable (line 147) | def readable(self) -> builtins.bool: method writable (line 151) | def writable(self) -> builtins.bool: method seekable (line 155) | def seekable(self) -> builtins.bool: method isatty (line 159) | def isatty(self) -> builtins.bool: method fileno (line 163) | def fileno(self) -> builtins.int: method close (line 167) | def close(self) -> None: method __enter__ (line 171) | def __enter__(self) -> _SlotLogWriter: method __exit__ (line 175) | def __exit__( class _TeeWriter (line 186) | class _TeeWriter: method inner (line 194) | def inner(self) -> typing.Any: method user_stream (line 199) | def user_stream(self) -> typing.Any: method name (line 204) | def name(self) -> builtins.str: method closed (line 209) | def closed(self) -> builtins.bool: method encoding (line 214) | def encoding(self) -> typing.Optional[builtins.str]: ... method newlines (line 216) | def newlines(self) -> typing.Optional[builtins.str]: ... method __new__ (line 217) | def __new__( method write (line 220) | def write(self, data: builtins.str) -> builtins.int: method flush (line 224) | def flush(self) -> None: method readable (line 228) | def readable(self) -> builtins.bool: ... method writable (line 229) | def writable(self) -> builtins.bool: ... method seekable (line 230) | def seekable(self) -> builtins.bool: ... method isatty (line 231) | def isatty(self) -> builtins.bool: ... method fileno (line 232) | def fileno(self) -> builtins.int: ... method close (line 233) | def close(self) -> None: ... method __enter__ (line 234) | def __enter__(self) -> _TeeWriter: ... method __exit__ (line 235) | def __exit__( function current_scope (line 242) | def current_scope() -> Scope: FILE: crates/coglet-python/src/audit.rs function install_audit_hook (line 41) | pub fn install_audit_hook(py: Python<'_>) -> PyResult<()> { function _coglet_audit_hook (line 70) | fn _coglet_audit_hook(py: Python<'_>, event: &str, args: &Bound<'_, PyAn... function handle_stream_replacement (line 108) | fn handle_stream_replacement(py: Python<'_>, name: &str, value: &Bound<'... function is_slot_log_writer (line 162) | pub(crate) fn is_slot_log_writer(py: Python<'_>, value: &Bound<'_, PyAny... function is_tee_writer (line 178) | pub(crate) fn is_tee_writer(value: &Bound<'_, PyAny>) -> bool { function get_inner_writer (line 191) | pub(crate) fn get_inner_writer(py: Python<'_>, tee: &Bound<'_, PyAny>) -... type _TeeWriter (line 215) | pub struct _TeeWriter { method new (line 234) | fn new(inner: Py, user_stream: Py, name: String) -> Self { method write (line 244) | fn write(&self, py: Python<'_>, data: &str) -> PyResult { method flush (line 261) | fn flush(&self, py: Python<'_>) -> PyResult<()> { method readable (line 267) | fn readable(&self) -> bool { method writable (line 271) | fn writable(&self) -> bool { method seekable (line 275) | fn seekable(&self) -> bool { method isatty (line 279) | fn isatty(&self, py: Python<'_>) -> PyResult { method fileno (line 284) | fn fileno(&self, py: Python<'_>) -> PyResult { method close (line 289) | fn close(&mut self) { method __enter__ (line 293) | fn __enter__(slf: PyRef<'_, Self>) -> PyRef<'_, Self> { method __exit__ (line 297) | fn __exit__( method encoding (line 307) | fn encoding(&self, py: Python<'_>) -> PyResult> { method newlines (line 315) | fn newlines(&self) -> Option { FILE: crates/coglet-python/src/bin/stub_gen.rs constant PUBLIC_REEXPORTS (line 17) | const PUBLIC_REEXPORTS: &[&str] = &["__build__", "__version__", "server"... constant PRIVATE_REEXPORTS (line 24) | const PRIVATE_REEXPORTS: &[&str] = &["_sdk"]; function main (line 26) | fn main() -> Result<()> { FILE: crates/coglet-python/src/cancel.rs function cancel_sync_thread (line 41) | pub fn cancel_sync_thread(py_thread_id: std::ffi::c_long) { function current_py_thread_id (line 82) | pub fn current_py_thread_id() -> std::ffi::c_long { FILE: crates/coglet-python/src/input.rs type PyObject (line 13) | type PyObject = Py; type PreparedInput (line 20) | pub struct PreparedInput { method new (line 29) | pub fn new(dict: Py, cleanup_paths: Vec) -> Self { method dict (line 37) | pub fn dict<'py>(&self, py: Python<'py>) -> Bound<'py, PyDict> { method drop (line 43) | fn drop(&mut self) { function prepare_input (line 76) | pub fn prepare_input( function detect_file_fields (line 91) | fn detect_file_fields(py: Python<'_>, func: &Bound<'_, PyAny>) -> PyResu... function coerce_url_strings (line 186) | fn coerce_url_strings( function download_url_paths_into_dict (line 244) | fn download_url_paths_into_dict( function file_fields_for (line 377) | fn file_fields_for(py_func_src: &str) -> HashSet { function detect_direct_file (line 399) | fn detect_direct_file() { function detect_list_file (line 407) | fn detect_list_file() { function detect_optional_file (line 414) | fn detect_optional_file() { function detect_file_union_none (line 426) | fn detect_file_union_none() { function detect_optional_list_file (line 438) | fn detect_optional_list_file() { function non_file_types_not_detected (line 450) | fn non_file_types_not_detected() { FILE: crates/coglet-python/src/lib.rs type BuildInfo (line 40) | pub struct BuildInfo { method __repr__ (line 56) | fn __repr__(&self) -> String { method new (line 69) | fn new() -> Self { method sha_display (line 80) | fn sha_display(&self) -> String { function set_active (line 89) | fn set_active() { function init_tracing (line 95) | fn init_tracing( function detect_version (line 152) | fn detect_version(py: Python<'_>, build: &BuildInfo) -> VersionInfo { function read_max_concurrency (line 175) | fn read_max_concurrency() -> usize { function read_setup_timeout (line 182) | fn read_setup_timeout() -> Option { type CogletServer (line 215) | pub struct CogletServer {} method active (line 222) | fn active(&self) -> bool { method serve (line 229) | fn serve( method run_worker (line 257) | fn run_worker(&self, py: Python<'_>) -> PyResult<()> { method __repr__ (line 282) | fn __repr__(&self) -> &'static str { function serve_impl (line 288) | fn serve_impl( function serve_subprocess (line 366) | fn serve_subprocess( function run_worker_with_init (line 479) | async fn run_worker_with_init() -> Result<(), String> { function coglet (line 542) | fn coglet(py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> { FILE: crates/coglet-python/src/log_writer.rs function get_sync_prediction_id_slot (line 47) | fn get_sync_prediction_id_slot() -> &'static Mutex> { function get_registry (line 56) | fn get_registry() -> &'static Mutex>> { function get_control_channel_sender_slot (line 60) | fn get_control_channel_sender_slot() -> &'static Mutex) -> Self { method try_send_log (line 84) | pub fn try_send_log(&self, source: LogSource, data: &str) { function register_control_channel_sender (line 111) | pub fn register_control_channel_sender(sender: Arc Option> { function get_prediction_contextvar (line 140) | pub fn get_prediction_contextvar(py: Python<'_>) -> PyResult<&'static Py... function register_prediction (line 166) | pub fn register_prediction(prediction_id: String, sender: Arc Option> { function set_current_prediction (line 201) | pub fn set_current_prediction(py: Python<'_>, prediction_id: &str) -> Py... function set_sync_prediction_id (line 210) | pub fn set_sync_prediction_id(prediction_id: Option<&str>) { function get_current_prediction_id (line 219) | fn get_current_prediction_id(py: Python<'_>) -> PyResult> { type SlotLogWriter (line 267) | pub struct SlotLogWriter { method write (line 292) | fn write(&self, py: Python<'_>, data: &str) -> PyResult { method emit_data (line 326) | fn emit_data(&self, py: Python<'_>, data: &str) -> PyResult<()> { method flush (line 369) | fn flush(&self, py: Python<'_>) -> PyResult<()> { method readable (line 385) | fn readable(&self) -> bool { method writable (line 390) | fn writable(&self) -> bool { method seekable (line 395) | fn seekable(&self) -> bool { method isatty (line 400) | fn isatty(&self, py: Python<'_>) -> PyResult { method fileno (line 407) | fn fileno(&self, py: Python<'_>) -> PyResult { method close (line 414) | fn close(&mut self) { method __enter__ (line 419) | fn __enter__(slf: PyRef<'_, Self>) -> PyRef<'_, Self> { method __exit__ (line 424) | fn __exit__( method encoding (line 435) | fn encoding(&self, py: Python<'_>) -> PyResult> { method newlines (line 444) | fn newlines(&self) -> Option { method buffer (line 450) | fn buffer(&self, py: Python<'_>) -> PyResult> { method new_stdout (line 461) | pub fn new_stdout(original: Py) -> Self { method new_stderr (line 471) | pub fn new_stderr(original: Py) -> Self { method write_outside_prediction (line 484) | fn write_outside_prediction(&self, _py: Python<'_>, data: &str) -> PyR... function install_slot_log_writers (line 511) | pub fn install_slot_log_writers(py: Python<'_>) -> PyResult { type PredictionLogGuard (line 541) | pub struct PredictionLogGuard { method enter (line 551) | pub fn enter(py: Python<'_>, prediction_id: String, sender: Arc &str { method drop (line 573) | fn drop(&mut self) { function registry_operations (line 595) | fn registry_operations() { function slot_sender_sends_log (line 610) | fn slot_sender_sends_log() { function slot_sender_ignores_empty (line 627) | fn slot_sender_ignores_empty() { function slot_sender_detects_closed_channel (line 638) | fn slot_sender_detects_closed_channel() { FILE: crates/coglet-python/src/metric_scope.rs type MetricValueType (line 29) | enum MetricValueType { method from_py (line 40) | fn from_py(obj: &Bound<'_, PyAny>) -> PyResult { method as_str (line 63) | fn as_str(&self) -> &'static str { type MetricRecorder (line 88) | pub struct MetricRecorder { method new (line 100) | pub fn new(sender: Arc) -> Self { method noop (line 109) | pub fn noop() -> Self { method record (line 129) | fn record( method delete (line 147) | fn delete(&self, key: &str) -> PyResult<()> { method __setitem__ (line 157) | fn __setitem__(&self, py: Python<'_>, key: &str, value: &Bound<'_, PyA... method __delitem__ (line 171) | fn __delitem__(&self, key: &str) -> PyResult<()> { method __repr__ (line 175) | fn __repr__(&self) -> String { type RecorderInner (line 92) | struct RecorderInner { type Scope (line 195) | pub struct Scope { method new (line 202) | pub fn new( method noop (line 218) | pub fn noop(py: Python<'_>) -> PyResult { method metrics (line 233) | fn metrics(&self, py: Python<'_>) -> Py { method context (line 241) | fn context(&self, py: Python<'_>) -> Py { method record_metric (line 249) | fn record_metric( method __repr__ (line 261) | fn __repr__(&self, py: Python<'_>) -> String { function parse_mode (line 271) | fn parse_mode(mode: Option<&str>) -> PyResult { function record_impl (line 283) | fn record_impl( function delete_impl (line 324) | fn delete_impl(inner: &mut RecorderInner, key: &str) -> PyResult<()> { function get_sync_scope_slot (line 348) | fn get_sync_scope_slot() -> &'static Mutex>> { function get_scope_contextvar (line 352) | fn get_scope_contextvar(py: Python<'_>) -> PyResult<&'static Py> { function set_current_scope (line 371) | pub fn set_current_scope(py: Python<'_>, scope: &Py) -> PyResult<... function set_sync_scope (line 378) | pub fn set_sync_scope(py: Python<'_>, scope: Option<&Py>) { function clear_sync_scope (line 386) | pub fn clear_sync_scope() { function py_current_scope (line 399) | pub fn py_current_scope(py: Python<'_>) -> PyResult> { type ScopeGuard (line 437) | pub struct ScopeGuard { method enter (line 445) | pub fn enter( method drop (line 460) | fn drop(&mut self) { function py_to_json (line 484) | fn py_to_json(obj: &Bound<'_, PyAny>) -> PyResult { FILE: crates/coglet-python/src/output.rs function process_output (line 22) | pub fn process_output<'py>( function process_output_item (line 31) | pub fn process_output_item<'py>( function make_encodeable (line 43) | fn make_encodeable<'py>(py: Python<'py>, obj: &Bound<'py, PyAny>) -> PyR... function encode_files (line 145) | fn encode_files<'py>(py: Python<'py>, obj: &Bound<'py, PyAny>) -> PyResu... function file_to_base64 (line 196) | fn file_to_base64<'py>(py: Python<'py>, fh: &Bound<'py, PyAny>) -> PyRes... function is_generator (line 239) | fn is_generator<'py>(py: Python<'py>, obj: &Bound<'py, PyAny>) -> PyResu... function encodeable (line 252) | fn encodeable(py_expr: &str) -> String { function processed (line 274) | fn processed(py_expr: &str) -> String { function encodeable_string (line 296) | fn encodeable_string() { function encodeable_int (line 301) | fn encodeable_int() { function encodeable_float (line 306) | fn encodeable_float() { function encodeable_bool (line 311) | fn encodeable_bool() { function encodeable_none (line 316) | fn encodeable_none() { function encodeable_list (line 323) | fn encodeable_list() { function encodeable_dict (line 328) | fn encodeable_dict() { function encodeable_tuple_to_list (line 336) | fn encodeable_tuple_to_list() { function encodeable_set_to_list (line 341) | fn encodeable_set_to_list() { function encodeable_frozenset_to_list (line 347) | fn encodeable_frozenset_to_list() { function encodeable_nested_dict (line 352) | fn encodeable_nested_dict() { function encodeable_enum (line 362) | fn encodeable_enum() { function encodeable_int_enum (line 370) | fn encodeable_int_enum() { function encodeable_datetime (line 382) | fn encodeable_datetime() { function encodeable_dataclass (line 391) | fn encodeable_dataclass() { function encodeable_nested_dataclass (line 406) | fn encodeable_nested_dataclass() { function encodeable_generator (line 423) | fn encodeable_generator() { function encodeable_enum_in_list (line 430) | fn encodeable_enum_in_list() { function encode_pathlike_to_base64 (line 446) | fn encode_pathlike_to_base64() { function encode_iobase_to_base64 (line 467) | fn encode_iobase_to_base64() { function encode_iobase_seeks_to_start (line 484) | fn encode_iobase_seeks_to_start() { function encode_file_in_dict (line 499) | fn encode_file_in_dict() { function encode_file_in_list (line 514) | fn encode_file_in_list() { function encode_string_passthrough (line 527) | fn encode_string_passthrough() { function encode_mime_type_guessing (line 533) | fn encode_mime_type_guessing() { function process_output_primitives_passthrough (line 550) | fn process_output_primitives_passthrough() { function process_output_dataclass_with_file (line 557) | fn process_output_dataclass_with_file() { FILE: crates/coglet-python/src/predictor.rs function get_collect_async_gen (line 30) | fn get_collect_async_gen(py: Python<'_>) -> Result, Prediction... function get_ctx_wrapper (line 56) | fn get_ctx_wrapper(py: Python<'_>) -> Result, PredictionError> { function is_cancelation_exception (line 79) | fn is_cancelation_exception(py: Python<'_>, err: &PyErr) -> bool { function format_validation_error (line 99) | fn format_validation_error(py: Python<'_>, err: &PyErr) -> String { function send_output_item (line 108) | fn send_output_item( type PyObject (line 194) | type PyObject = Py; type PredictKind (line 198) | pub enum PredictKind { type TrainKind (line 209) | pub enum TrainKind { type PredictorKind (line 220) | pub enum PredictorKind { type PythonPredictor (line 235) | pub struct PythonPredictor { method load (line 248) | pub fn load(py: Python<'_>, predictor_ref: &str) -> PyResult { method unwrap_field_info_defaults (line 337) | fn unwrap_field_info_defaults( method detect_async (line 406) | fn detect_async( method is_async (line 436) | pub fn is_async(&self) -> bool { method has_train (line 448) | pub fn has_train(&self) -> bool { method is_train_async (line 456) | pub fn is_train_async(&self) -> bool { method setup (line 470) | pub fn setup(&self, py: Python<'_>) -> PyResult<()> { method predict_func (line 498) | pub fn predict_func<'py>(&self, py: Python<'py>) -> PyResult(&self, py: Python<'py>) -> PyResult, input: &Bound<'_, PyDict>) -... method train_raw (line 536) | pub fn train_raw(&self, py: Python<'_>, input: &Bound<'_, PyDict>) -> ... method call_method_raw (line 548) | fn call_method_raw( method predict_worker (line 577) | pub fn predict_worker( method train_worker (line 650) | pub fn train_worker( method process_generator_output (line 722) | fn process_generator_output( method process_single_output (line 752) | fn process_single_output( method predict_async_worker (line 858) | pub fn predict_async_worker( method process_async_result (line 949) | pub fn process_async_result( method train_async_worker (line 995) | pub fn train_async_worker( constant HEALTHCHECK_TIMEOUT (line 1070) | const HEALTHCHECK_TIMEOUT: f64 = 5.0; method has_healthcheck (line 1073) | pub fn has_healthcheck(&self, py: Python<'_>) -> bool { method is_healthcheck_async (line 1084) | pub fn is_healthcheck_async(&self, py: Python<'_>) -> bool { method healthcheck_sync (line 1110) | pub fn healthcheck_sync(&self, py: Python<'_>) -> coglet_core::orchest... method healthcheck_async (line 1172) | pub fn healthcheck_async( FILE: crates/coglet-python/src/worker_bridge.rs type HandlerMode (line 16) | pub enum HandlerMode { type SdkImplementation (line 28) | pub enum SdkImplementation { method fmt (line 36) | fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { type SlotState (line 46) | pub enum SlotState { method is_cancelled (line 65) | pub fn is_cancelled(&self) -> bool { method mark_cancelled (line 73) | pub fn mark_cancelled(&mut self) { type PythonPredictHandler (line 92) | pub struct PythonPredictHandler { method new (line 108) | pub fn new(predictor_ref: String) -> Result { method new_train (line 125) | pub fn new_train(predictor_ref: String) -> Result { method init_async_loop (line 138) | fn init_async_loop() -> Result<(Py, JoinHandle<()>), SetupError> { method is_cancelled (line 183) | fn is_cancelled(&self, slot: SlotId) -> bool { method take_cancelled (line 189) | fn take_cancelled(&self, slot: SlotId) -> bool { method start_sync_prediction (line 204) | fn start_sync_prediction(&self, slot: SlotId, py_thread_id: std::ffi::... method start_async_prediction (line 216) | fn start_async_prediction(&self, slot: SlotId, future: Py) { method finish_prediction (line 228) | fn finish_prediction(&self, slot: SlotId) { method cancel_async_future (line 235) | fn cancel_async_future(&self, slot: SlotId) -> bool { method get_async_loop (line 265) | fn get_async_loop(&self) -> Option> { method setup (line 278) | async fn setup(&self) -> Result<(), SetupError> { method predict (line 307) | async fn predict( method cancel (line 575) | fn cancel(&self, slot: SlotId) { method healthcheck (line 607) | async fn healthcheck(&self) -> coglet_core::orchestrator::HealthcheckRes... method drop (line 652) | fn drop(&mut self) { function output_to_json (line 692) | fn output_to_json(output: coglet_core::PredictionOutput) -> serde_json::... FILE: crates/coglet-python/tests/test_coglet.py class TestModuleStructure (line 21) | class TestModuleStructure: method test_version_is_pep440 (line 24) | def test_version_is_pep440(self) -> None: method test_version_is_str (line 32) | def test_version_is_str(self) -> None: method test_build_info_exists (line 35) | def test_build_info_exists(self) -> None: method test_build_info_fields_are_strings (line 42) | def test_build_info_fields_are_strings(self) -> None: method test_build_info_version_matches_module_version (line 49) | def test_build_info_version_matches_module_version(self) -> None: method test_build_info_repr (line 52) | def test_build_info_repr(self) -> None: method test_build_info_frozen (line 58) | def test_build_info_frozen(self) -> None: method test_server_exists (line 62) | def test_server_exists(self) -> None: method test_server_active_is_false (line 65) | def test_server_active_is_false(self) -> None: method test_server_active_is_property (line 69) | def test_server_active_is_property(self) -> None: method test_server_frozen (line 73) | def test_server_frozen(self) -> None: method test_server_active_not_settable (line 77) | def test_server_active_not_settable(self) -> None: method test_server_repr (line 81) | def test_server_repr(self) -> None: method test_sdk_submodule_exists (line 84) | def test_sdk_submodule_exists(self) -> None: method test_sdk_has_slot_log_writer (line 87) | def test_sdk_has_slot_log_writer(self) -> None: method test_sdk_has_tee_writer (line 90) | def test_sdk_has_tee_writer(self) -> None: method test_all_excludes_internals (line 93) | def test_all_excludes_internals(self) -> None: function sync_predictor (line 104) | def sync_predictor(tmp_path: Path) -> Path: function generator_predictor (line 128) | def generator_predictor(tmp_path: Path) -> Path: function async_predictor (line 154) | def async_predictor(tmp_path: Path) -> Path: function async_generator_predictor (line 181) | def async_generator_predictor(tmp_path: Path) -> Path: class CogletServer (line 208) | class CogletServer: method __init__ (line 211) | def __init__(self, predictor_path: Path, port: int = 0): method __enter__ (line 220) | def __enter__(self): method __exit__ (line 247) | def __exit__(self, exc_type, exc_val, exc_tb): method _read_stderr (line 252) | def _read_stderr(self): method _discover_port (line 261) | def _discover_port(self, timeout: float = 5.0): method _wait_for_ready (line 285) | def _wait_for_ready(self, timeout: float = 10.0): method base_url (line 309) | def base_url(self) -> str: method health_check (line 312) | def health_check(self) -> dict: method predict (line 317) | def predict(self, input_data: dict) -> dict: class TestHealthCheck (line 325) | class TestHealthCheck: method test_returns_ready_status (line 328) | def test_returns_ready_status(self, sync_predictor: Path): method test_returns_version_info (line 333) | def test_returns_version_info(self, sync_predictor: Path): class TestSyncPredictor (line 342) | class TestSyncPredictor: method test_basic_prediction (line 345) | def test_basic_prediction(self, sync_predictor: Path): method test_default_input (line 351) | def test_default_input(self, sync_predictor: Path): method test_includes_predict_time (line 357) | def test_includes_predict_time(self, sync_predictor: Path): class TestGeneratorPredictor (line 365) | class TestGeneratorPredictor: method test_returns_array_output (line 368) | def test_returns_array_output(self, generator_predictor: Path): method test_custom_count (line 374) | def test_custom_count(self, generator_predictor: Path): class TestAsyncPredictor (line 380) | class TestAsyncPredictor: method test_basic_prediction (line 383) | def test_basic_prediction(self, async_predictor: Path): method test_sequential_requests (line 389) | def test_sequential_requests(self, async_predictor: Path): class TestAsyncGeneratorPredictor (line 402) | class TestAsyncGeneratorPredictor: method test_returns_array_output (line 405) | def test_returns_array_output(self, async_generator_predictor: Path): function slow_sync_predictor (line 417) | def slow_sync_predictor(tmp_path: Path) -> Path: function blocking_sleep_predictor (line 445) | def blocking_sleep_predictor(tmp_path: Path) -> Path: function slow_async_predictor (line 471) | def slow_async_predictor(tmp_path: Path) -> Path: function _wait_for_health_status (line 495) | def _wait_for_health_status( class TestCancellation (line 515) | class TestCancellation: method test_cancel_endpoint_returns_404_for_unknown_id (line 518) | def test_cancel_endpoint_returns_404_for_unknown_id(self, sync_predict... method test_prediction_response_includes_id (line 526) | def test_prediction_response_includes_id(self, sync_predictor: Path): method test_cancel_running_sync_prediction (line 533) | def test_cancel_running_sync_prediction(self, slow_sync_predictor: Path): method test_cancel_running_async_prediction (line 557) | def test_cancel_running_async_prediction(self, slow_async_predictor: P... method test_repeated_cancel_is_idempotent (line 591) | def test_repeated_cancel_is_idempotent( method test_cancel_sync_prediction_connection_drop (line 631) | def test_cancel_sync_prediction_connection_drop(self, slow_sync_predic... FILE: crates/coglet/src/bridge/codec.rs type JsonCodec (line 16) | pub struct JsonCodec { method default (line 22) | fn default() -> Self { function new (line 28) | pub fn new() -> Self { type Item (line 39) | type Item = T; type Error (line 40) | type Error = io::Error; method decode (line 42) | fn decode(&mut self, src: &mut BytesMut) -> Result, S... type Error (line 55) | type Error = io::Error; function encode (line 57) | fn encode(&mut self, item: T, dst: &mut BytesMut) -> Result<(), Self::Er... function codec_roundtrip_control_request (line 88) | fn codec_roundtrip_control_request() { function codec_roundtrip_control_response (line 101) | fn codec_roundtrip_control_response() { function codec_roundtrip_slot_request (line 117) | fn codec_roundtrip_slot_request() { function codec_roundtrip_slot_response (line 158) | fn codec_roundtrip_slot_response() { FILE: crates/coglet/src/bridge/protocol.rs type SlotId (line 18) | pub struct SlotId(uuid::Uuid); method new (line 21) | pub fn new() -> Self { method as_uuid (line 25) | pub fn as_uuid(&self) -> &uuid::Uuid { method parse (line 29) | pub fn parse(s: &str) -> Result { method fmt (line 42) | fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { method default (line 36) | fn default() -> Self { constant MAX_INLINE_IPC_SIZE (line 51) | pub const MAX_INLINE_IPC_SIZE: usize = 1024 * 1024 * 6; constant MAX_WORKER_LOG_SIZE (line 53) | const MAX_WORKER_LOG_SIZE: usize = 1024 * 1024 * 4; constant WORKER_LOG_TRUNCATE_NOTICE (line 54) | const WORKER_LOG_TRUNCATE_NOTICE: &str = "[**** LOG LINE TRUNCATED AT 4 ... function truncate_worker_log (line 58) | pub fn truncate_worker_log(mut log_message: String) -> String { type ControlRequest (line 71) | pub enum ControlRequest { type ControlResponse (line 96) | pub enum ControlResponse { type HealthcheckStatus (line 160) | pub enum HealthcheckStatus { type SlotOutcome (line 169) | pub enum SlotOutcome { method idle (line 175) | pub fn idle(slot: SlotId) -> Self { method poisoned (line 179) | pub fn poisoned(slot: SlotId, error: impl Into) -> Self { method slot_id (line 186) | pub fn slot_id(&self) -> SlotId { method is_poisoned (line 193) | pub fn is_poisoned(&self) -> bool { method into_control_response (line 197) | pub fn into_control_response(self) -> ControlResponse { type SlotRequest (line 208) | pub enum SlotRequest { method prediction_id (line 230) | pub fn prediction_id(&self) -> &str { method rehydrate_input (line 240) | pub fn rehydrate_input( type FileOutputKind (line 278) | pub enum FileOutputKind { type MetricMode (line 288) | pub enum MetricMode { type SlotResponse (line 300) | pub enum SlotResponse { type LogSource (line 359) | pub enum LogSource { function test_slot_id (line 370) | fn test_slot_id() -> SlotId { function control_init_serializes (line 375) | fn control_init_serializes() { function control_cancel_serializes (line 390) | fn control_cancel_serializes() { function control_shutdown_serializes (line 398) | fn control_shutdown_serializes() { function control_healthcheck_serializes (line 404) | fn control_healthcheck_serializes() { function control_healthcheck_result_healthy_serializes (line 412) | fn control_healthcheck_result_healthy_serializes() { function control_healthcheck_result_unhealthy_serializes (line 422) | fn control_healthcheck_result_unhealthy_serializes() { function control_ready_serializes (line 432) | fn control_ready_serializes() { function control_ready_with_schema_serializes (line 441) | fn control_ready_with_schema_serializes() { function control_idle_serializes (line 453) | fn control_idle_serializes() { function control_cancelled_serializes (line 461) | fn control_cancelled_serializes() { function control_failed_serializes (line 469) | fn control_failed_serializes() { function slot_predict_serializes (line 478) | fn slot_predict_serializes() { function slot_predict_file_input_serializes (line 490) | fn slot_predict_file_input_serializes() { function slot_log_serializes (line 502) | fn slot_log_serializes() { function slot_output_serializes (line 511) | fn slot_output_serializes() { function slot_done_serializes (line 519) | fn slot_done_serializes() { function slot_failed_serializes (line 530) | fn slot_failed_serializes() { function slot_cancelled_serializes (line 539) | fn slot_cancelled_serializes() { function slot_metric_replace_serializes (line 547) | fn slot_metric_replace_serializes() { function slot_metric_increment_serializes (line 557) | fn slot_metric_increment_serializes() { function slot_metric_append_serializes (line 567) | fn slot_metric_append_serializes() { function slot_metric_delete_serializes (line 577) | fn slot_metric_delete_serializes() { function slot_metric_complex_value_serializes (line 587) | fn slot_metric_complex_value_serializes() { function rehydrate_input_inline (line 597) | fn rehydrate_input_inline() { function rehydrate_input_from_file (line 612) | fn rehydrate_input_from_file() { function rehydrate_input_neither_errors (line 633) | fn rehydrate_input_neither_errors() { function rehydrate_input_corrupt_file_errors (line 646) | fn rehydrate_input_corrupt_file_errors() { function truncate_worker_log_truncates_long_messages (line 663) | fn truncate_worker_log_truncates_long_messages() { function truncate_worker_log_does_not_truncate_short_messages (line 676) | fn truncate_worker_log_does_not_truncate_short_messages() { FILE: crates/coglet/src/bridge/transport.rs type ChildTransportInfo (line 15) | pub enum ChildTransportInfo { type NamedSocketTransport (line 30) | pub struct NamedSocketTransport { method create (line 39) | pub async fn create(num_slots: usize) -> io::Result<(Self, ChildTransp... method accept_connections (line 80) | pub async fn accept_connections(&mut self, num_slots: usize) -> io::Re... method connect (line 94) | pub async fn connect(dir: PathBuf, num_slots: usize) -> io::Result Option<&mut UnixStream> { method drain_sockets (line 120) | pub fn drain_sockets(&mut self) -> Vec { method dir (line 124) | pub fn dir(&self) -> &PathBuf { method num_slots (line 128) | pub fn num_slots(&self) -> usize { method cleanup (line 132) | pub fn cleanup(&mut self) -> io::Result<()> { method drop (line 142) | fn drop(&mut self) { type AbstractSocketTransport (line 153) | pub struct AbstractSocketTransport { method create (line 163) | pub async fn create(num_slots: usize) -> io::Result<(Self, ChildTransp... method accept_connections (line 197) | pub async fn accept_connections(&mut self, num_slots: usize) -> io::Re... method connect (line 211) | pub async fn connect(prefix: String, num_slots: usize) -> io::Result Option<&mut UnixStream> { method drain_sockets (line 243) | pub fn drain_sockets(&mut self) -> Vec { method num_slots (line 247) | pub fn num_slots(&self) -> usize { type SlotTransport (line 252) | pub enum SlotTransport { method slot_socket (line 259) | pub fn slot_socket(&mut self, slot: usize) -> Option<&mut UnixStream> { method drain_sockets (line 267) | pub fn drain_sockets(&mut self) -> Vec { method num_slots (line 275) | pub fn num_slots(&self) -> usize { method accept_connections (line 283) | pub async fn accept_connections(&mut self, num_slots: usize) -> io::Re... function create_transport (line 293) | pub async fn create_transport(num_slots: usize) -> io::Result<(SlotTrans... function connect_transport (line 307) | pub async fn connect_transport(info: ChildTransportInfo) -> io::Result Self { type SetupStatus (line 55) | pub enum SetupStatus { type SetupResult (line 63) | pub struct SetupResult { method starting (line 79) | pub fn starting() -> Self { method succeeded (line 89) | pub fn succeeded(mut self, logs: String) -> Self { method failed (line 97) | pub fn failed(mut self, logs: String) -> Self { function health_default_is_unknown (line 110) | fn health_default_is_unknown() { function health_serializes_screaming_snake_case (line 115) | fn health_serializes_screaming_snake_case() { function health_response_serializes_screaming_snake_case (line 130) | fn health_response_serializes_screaming_snake_case() { function health_deserializes_screaming_snake_case (line 146) | fn health_deserializes_screaming_snake_case() { function setup_status_serializes_lowercase (line 158) | fn setup_status_serializes_lowercase() { function setup_status_deserializes_lowercase (line 170) | fn setup_status_deserializes_lowercase() { FILE: crates/coglet/src/input_validation.rs type ValidationError (line 13) | pub struct ValidationError { type InputValidator (line 23) | pub struct InputValidator { method from_openapi_schema (line 38) | pub fn from_openapi_schema(schema: &Value) -> Option { method from_openapi_schema_key (line 46) | pub fn from_openapi_schema_key(schema: &Value, key: &str) -> Option usize { method validate (line 98) | pub fn validate(&self, input: &Value) -> Result<(), Vec) { function resolve_ref (line 210) | fn resolve_ref(ref_str: &str, all_schemas: Option<&Value>) -> Option Value { function validates_required_fields (line 231) | fn validates_required_fields() { function rejects_additional_properties (line 253) | fn rejects_additional_properties() { function missing_and_extra_fields (line 274) | fn missing_and_extra_fields() { function validates_types (line 297) | fn validates_types() { function no_schema_returns_none (line 317) | fn no_schema_returns_none() { function resolves_ref_for_choices (line 323) | fn resolves_ref_for_choices() { function optional_fields_work (line 356) | fn optional_fields_work() { FILE: crates/coglet/src/orchestrator.rs function upload_file (line 38) | async fn upload_file( function ensure_trailing_slash (line 77) | fn ensure_trailing_slash(s: &str) -> String { function try_lock_prediction (line 88) | fn try_lock_prediction( function wrap_outputs (line 112) | fn wrap_outputs( function emit_worker_log (line 133) | fn emit_worker_log(target: &str, level: &str, msg: &str) { type HealthcheckResult (line 208) | pub struct HealthcheckResult { method healthy (line 214) | pub fn healthy() -> Self { method unhealthy (line 221) | pub fn unhealthy(error: impl Into) -> Self { method is_healthy (line 228) | pub fn is_healthy(&self) -> bool { type Orchestrator (line 239) | pub trait Orchestrator: Send + Sync { method register_prediction (line 241) | async fn register_prediction( method cancel_by_prediction_id (line 252) | async fn cancel_by_prediction_id(&self, prediction_id: &str) -> Result... method healthcheck (line 255) | async fn healthcheck(&self) -> Result Result<(), OrchestratorError>; method register_prediction (line 372) | async fn register_prediction( method cancel_by_prediction_id (line 384) | async fn cancel_by_prediction_id(&self, prediction_id: &str) -> Result... method healthcheck (line 391) | async fn healthcheck(&self) -> Result Result<(), OrchestratorError> { type WorkerSpawnConfig (line 262) | pub struct WorkerSpawnConfig { type SpawnError (line 267) | pub enum SpawnError { type WorkerSpawner (line 275) | pub trait WorkerSpawner: Send + Sync { method spawn (line 276) | fn spawn(&self, config: &WorkerSpawnConfig) -> Result; method spawn (line 283) | fn spawn(&self, _config: &WorkerSpawnConfig) -> Result) -> Self { method with_upload_url (line 318) | pub fn with_upload_url(mut self, upload_url: Option) -> Self { method with_num_slots (line 323) | pub fn with_num_slots(mut self, n: usize) -> Self { method with_train (line 328) | pub fn with_train(mut self, is_train: bool) -> Self { method with_async (line 333) | pub fn with_async(mut self, is_async: bool) -> Self { method with_setup_timeout (line 338) | pub fn with_setup_timeout(mut self, timeout: Option) -> Self { method with_spawner (line 343) | pub fn with_spawner(mut self, spawner: Arc) -> Self { type OrchestratorReady (line 349) | pub struct OrchestratorReady { type OrchestratorHandle (line 356) | pub struct OrchestratorHandle { method cancel (line 434) | pub async fn cancel(&self, slot_id: SlotId) -> Result<(), Orchestrator... method slot_ids (line 442) | pub fn slot_ids(&self) -> &[SlotId] { method wait (line 446) | pub async fn wait(&mut self) -> Result<(), OrchestratorError> { type OrchestratorError (line 455) | pub enum OrchestratorError { function spawn_worker (line 468) | pub async fn spawn_worker( function run_event_loop (line 692) | async fn run_event_loop( function wrap_outputs_schema_array_empty (line 1241) | fn wrap_outputs_schema_array_empty() { function wrap_outputs_schema_array_single_item (line 1249) | fn wrap_outputs_schema_array_single_item() { function wrap_outputs_schema_array_multiple_items (line 1260) | fn wrap_outputs_schema_array_multiple_items() { function wrap_outputs_schema_array_overrides_is_stream_false (line 1274) | fn wrap_outputs_schema_array_overrides_is_stream_false() { function wrap_outputs_predictor_stream_empty (line 1284) | fn wrap_outputs_predictor_stream_empty() { function wrap_outputs_predictor_stream_single_item (line 1292) | fn wrap_outputs_predictor_stream_single_item() { function wrap_outputs_predictor_stream_multiple_items (line 1300) | fn wrap_outputs_predictor_stream_multiple_items() { function wrap_outputs_scalar_empty (line 1311) | fn wrap_outputs_scalar_empty() { function wrap_outputs_scalar_single (line 1319) | fn wrap_outputs_scalar_single() { function wrap_outputs_scalar_multiple_falls_back_to_stream (line 1330) | fn wrap_outputs_scalar_multiple_falls_back_to_stream() { function done_is_stream_false_omitted_from_json (line 1342) | fn done_is_stream_false_omitted_from_json() { function done_is_stream_true_present_in_json (line 1357) | fn done_is_stream_true_present_in_json() { function done_without_is_stream_deserializes_as_false (line 1369) | fn done_without_is_stream_deserializes_as_false() { FILE: crates/coglet/src/permit/pool.rs type PermitInner (line 18) | pub(crate) struct PermitInner { type PoolConnection (line 25) | struct PoolConnection { method clone (line 31) | fn clone(&self) -> Self { type PermitInUse (line 40) | pub struct PermitInUse { method new (line 49) | pub(crate) fn new( method slot_id (line 68) | pub fn slot_id(&self) -> SlotId { method into_idle (line 74) | pub fn into_idle(mut self) -> PermitIdle { method into_poisoned (line 88) | pub fn into_poisoned(mut self) -> PermitPoisoned { method send (line 96) | pub async fn send(&mut self, request: SlotRequest) -> Result<(), Permi... method drop (line 106) | fn drop(&mut self) { type PermitIdle (line 115) | pub struct PermitIdle { method slot_id (line 124) | pub fn slot_id(&self) -> SlotId { method drop (line 130) | fn drop(&mut self) { type PermitPoisoned (line 153) | pub struct PermitPoisoned { method slot_id (line 159) | pub fn slot_id(&self) -> SlotId { method drop (line 165) | fn drop(&mut self) { type AnyPermit (line 171) | pub enum AnyPermit { method slot_id (line 178) | pub fn slot_id(&self) -> SlotId { method is_idle (line 186) | pub fn is_idle(&self) -> bool { method is_poisoned (line 190) | pub fn is_poisoned(&self) -> bool { method is_in_use (line 194) | pub fn is_in_use(&self) -> bool { type InactiveSlotIdleToken (line 201) | pub struct InactiveSlotIdleToken { method new (line 206) | pub fn new(slot_id: SlotId) -> Self { method slot_id (line 210) | pub fn slot_id(&self) -> SlotId { method activate (line 214) | pub fn activate(self) -> SlotIdleToken { type SlotIdleToken (line 231) | pub struct SlotIdleToken { constant ALERT_THRESHOLD (line 238) | const ALERT_THRESHOLD: std::time::Duration = std::time::Duration::from... method slot_id (line 240) | pub fn slot_id(&self) -> SlotId { method consume (line 244) | pub fn consume(self) { method drop (line 254) | fn drop(&mut self) { type PermitError (line 260) | pub enum PermitError { type PermitPool (line 271) | pub struct PermitPool { method new (line 281) | pub fn new(num_slots: usize) -> Self { method add_permit (line 293) | pub fn add_permit( method poison (line 324) | pub fn poison(&self, slot_id: SlotId) { method is_poisoned (line 339) | pub fn is_poisoned(&self, slot_id: SlotId) -> bool { method try_acquire (line 350) | pub fn try_acquire(&self) -> Option { method acquire (line 370) | pub async fn acquire(&self) -> Option { method num_slots (line 390) | pub fn num_slots(&self) -> usize { method available (line 394) | pub fn available(&self) -> usize { function make_socket_pair (line 404) | async fn make_socket_pair() -> (OwnedWriteHalf, tokio::net::unix::OwnedR... function pool_add_and_acquire (line 412) | async fn pool_add_and_acquire() { function permit_returns_to_pool_when_idle (line 435) | async fn permit_returns_to_pool_when_idle() { function permit_orphaned_when_poisoned (line 453) | async fn permit_orphaned_when_poisoned() { function pool_poison_idle_slot (line 471) | async fn pool_poison_idle_slot() { function pool_poison_in_use_slot_prevents_return (line 498) | async fn pool_poison_in_use_slot_prevents_return() { function pool_poison_is_idempotent (line 520) | async fn pool_poison_is_idempotent() { FILE: crates/coglet/src/permit/slot.rs type SlotError (line 17) | pub enum SlotError { type UnregisteredPredictionSlot (line 26) | pub struct UnregisteredPredictionSlot { method new (line 32) | pub fn new( method into_parts (line 43) | pub fn into_parts(self) -> (tokio::sync::oneshot::Sender Arc> { type PredictionSlot (line 55) | pub struct PredictionSlot { method new (line 63) | pub fn new( method prediction (line 77) | pub fn prediction(&self) -> Arc> { method permit_mut (line 81) | pub fn permit_mut(&mut self) -> Option<&mut PermitInUse> { method slot_id (line 88) | pub fn slot_id(&self) -> SlotId { method into_idle (line 97) | pub async fn into_idle(mut self) -> Result<(), SlotError> { method is_idle (line 138) | pub fn is_idle(&self) -> bool { method id (line 142) | pub fn id(&self) -> String { method drop (line 151) | fn drop(&mut self) { function slot_creation (line 175) | async fn slot_creation() { function slot_mark_idle_returns_permit (line 193) | async fn slot_mark_idle_returns_permit() { function slot_not_idle_orphans_permit (line 217) | async fn slot_not_idle_orphans_permit() { function slot_idle_channel_closed_does_not_return_permit (line 237) | async fn slot_idle_channel_closed_does_not_return_permit() { FILE: crates/coglet/src/prediction.rs type PredictionStatus (line 14) | pub enum PredictionStatus { method is_terminal (line 23) | pub fn is_terminal(&self) -> bool { method as_str (line 27) | pub fn as_str(&self) -> &'static str { type PredictionOutput (line 41) | pub enum PredictionOutput { method is_stream (line 47) | pub fn is_stream(&self) -> bool { method into_values (line 51) | pub fn into_values(self) -> Vec { method final_value (line 59) | pub fn final_value(&self) -> &serde_json::Value { type Prediction (line 68) | pub struct Prediction { method new (line 84) | pub fn new(id: String, webhook: Option) -> Self { method id (line 100) | pub fn id(&self) -> &str { method cancel_token (line 104) | pub fn cancel_token(&self) -> CancellationToken { method is_canceled (line 108) | pub fn is_canceled(&self) -> bool { method status (line 112) | pub fn status(&self) -> PredictionStatus { method is_terminal (line 116) | pub fn is_terminal(&self) -> bool { method set_processing (line 120) | pub fn set_processing(&mut self) { method set_succeeded (line 125) | pub fn set_succeeded(&mut self, output: PredictionOutput) { method set_failed (line 141) | pub fn set_failed(&mut self, error: String) { method set_canceled (line 151) | pub fn set_canceled(&mut self) { method elapsed (line 160) | pub fn elapsed(&self) -> std::time::Duration { method append_log (line 164) | pub fn append_log(&mut self, data: &str) { method logs (line 169) | pub fn logs(&self) -> &str { method set_metric (line 180) | pub fn set_metric(&mut self, name: String, value: serde_json::Value, m... method set_metric_dotpath (line 227) | fn set_metric_dotpath(&mut self, parts: &[&str], value: serde_json::Va... method metrics (line 293) | pub fn metrics(&self) -> &HashMap { method append_output (line 297) | pub fn append_output(&mut self, output: serde_json::Value) { method outputs (line 302) | pub fn outputs(&self) -> &[serde_json::Value] { method take_outputs (line 306) | pub fn take_outputs(&mut self) -> Vec { method output (line 310) | pub fn output(&self) -> Option<&PredictionOutput> { method error (line 314) | pub fn error(&self) -> Option<&str> { method wait (line 318) | pub async fn wait(&self) { method completion (line 325) | pub fn completion(&self) -> Arc { method take_webhook (line 330) | pub fn take_webhook(&mut self) -> Option { method fire_webhook (line 338) | fn fire_webhook(&self, event: WebhookEventType) { method fire_terminal_webhook (line 349) | fn fire_terminal_webhook(&mut self) { method build_state_snapshot (line 365) | pub fn build_state_snapshot(&self) -> serde_json::Value { method build_webhook_payload (line 401) | fn build_webhook_payload(&self) -> serde_json::Value { method build_terminal_response (line 405) | pub fn build_terminal_response(&self) -> serde_json::Value { function status_is_terminal (line 415) | fn status_is_terminal() { function new_starts_in_starting_status (line 424) | fn new_starts_in_starting_status() { function set_succeeded (line 431) | fn set_succeeded() { function set_failed (line 438) | fn set_failed() { function set_canceled (line 445) | fn set_canceled() { function cancel_token_works (line 452) | fn cancel_token_works() { function elapsed_time_increases (line 462) | fn elapsed_time_increases() { function append_log (line 471) | fn append_log() { function append_output (line 479) | fn append_output() { function wait_returns_immediately_if_terminal (line 487) | async fn wait_returns_immediately_if_terminal() { function prediction_output_single (line 496) | fn prediction_output_single() { function prediction_output_stream (line 503) | fn prediction_output_stream() { function metric_replace_sets_value (line 513) | fn metric_replace_sets_value() { function metric_replace_overwrites (line 520) | fn metric_replace_overwrites() { function metric_replace_null_deletes (line 528) | fn metric_replace_null_deletes() { function metric_increment_integers (line 536) | fn metric_increment_integers() { function metric_increment_floats (line 544) | fn metric_increment_floats() { function metric_increment_creates_from_zero (line 560) | fn metric_increment_creates_from_zero() { function metric_append_creates_array (line 567) | fn metric_append_creates_array() { function metric_append_to_non_array_wraps (line 583) | fn metric_append_to_non_array_wraps() { function metric_dotpath_creates_nested (line 591) | fn metric_dotpath_creates_nested() { function metric_dotpath_deep (line 605) | fn metric_dotpath_deep() { function metric_dotpath_multiple_leaves (line 612) | fn metric_dotpath_multiple_leaves() { function metric_dotpath_delete_leaf (line 631) | fn metric_dotpath_delete_leaf() { function metric_dotpath_increment (line 648) | fn metric_dotpath_increment() { function metric_complex_values (line 664) | fn metric_complex_values() { function terminal_snapshot_merges_metrics_with_predict_time (line 684) | fn terminal_snapshot_merges_metrics_with_predict_time() { function terminal_snapshot_predict_time_overrides_user (line 698) | fn terminal_snapshot_predict_time_overrides_user() { function terminal_state_guard_set_failed_after_succeeded (line 715) | fn terminal_state_guard_set_failed_after_succeeded() { function terminal_state_guard_set_succeeded_after_failed (line 725) | fn terminal_state_guard_set_succeeded_after_failed() { function terminal_state_guard_set_canceled_after_succeeded (line 734) | fn terminal_state_guard_set_canceled_after_succeeded() { FILE: crates/coglet/src/predictor.rs type PredictionResult (line 10) | pub struct PredictionResult { type PredictionMetrics (line 20) | pub struct PredictionMetrics { type PredictionGuard (line 25) | pub struct PredictionGuard { method new (line 32) | pub fn new() -> Self { method cancel_token (line 40) | pub fn cancel_token(&self) -> CancellationToken { method is_cancelled (line 44) | pub fn is_cancelled(&self) -> bool { method cancel (line 48) | pub fn cancel(&self) { method finish (line 52) | pub fn finish(mut self) -> PredictionMetrics { method default (line 59) | fn default() -> Self { type PredictionError (line 65) | pub enum PredictionError { function prediction_output_single_is_not_stream (line 87) | fn prediction_output_single_is_not_stream() { function prediction_output_stream_is_stream (line 93) | fn prediction_output_stream_is_stream() { function prediction_output_serializes_untagged (line 99) | fn prediction_output_serializes_untagged() { function prediction_guard_tracks_time (line 108) | fn prediction_guard_tracks_time() { function prediction_error_display (line 120) | fn prediction_error_display() { FILE: crates/coglet/src/service.rs function try_lock_prediction (line 29) | fn try_lock_prediction( type CreatePredictionError (line 46) | pub enum CreatePredictionError { type HealthSnapshot (line 55) | pub struct HealthSnapshot { method is_ready (line 64) | pub fn is_ready(&self) -> bool { method is_busy (line 69) | pub fn is_busy(&self) -> bool { type PredictionEntry (line 78) | struct PredictionEntry { type PredictionHandle (line 85) | pub struct PredictionHandle { method id (line 91) | pub fn id(&self) -> &str { method cancel_token (line 95) | pub fn cancel_token(&self) -> CancellationToken { method sync_guard (line 104) | pub fn sync_guard(&self, service: Arc) -> SyncPredi... type SyncPredictionGuard (line 115) | pub struct SyncPredictionGuard { method new (line 121) | pub fn new(prediction_id: String, service: Arc) -> ... method disarm (line 128) | pub fn disarm(&mut self) { method drop (line 134) | fn drop(&mut self) { type OrchestratorState (line 144) | pub struct OrchestratorState { method clone (line 150) | fn clone(&self) -> Self { type PredictionService (line 162) | pub struct PredictionService { method new_no_pool (line 186) | pub fn new_no_pool() -> Self { method set_orchestrator (line 203) | pub async fn set_orchestrator( method has_orchestrator (line 211) | pub async fn has_orchestrator(&self) -> bool { method shutdown (line 219) | pub async fn shutdown(&self) { method with_health (line 231) | pub fn with_health(mut self, health: Health) -> Self { method with_version (line 238) | pub fn with_version(mut self, version: VersionInfo) -> Self { method version (line 244) | pub fn version(&self) -> &VersionInfo { method supports_training (line 249) | pub async fn supports_training(&self) -> bool { method pool (line 254) | pub async fn pool(&self) -> Option> { method health (line 262) | pub async fn health(&self) -> HealthSnapshot { method set_health (line 291) | pub async fn set_health(&self, health: Health) { method set_setup_result (line 301) | pub async fn set_setup_result(&self, result: SetupResult) { method set_schema (line 312) | pub async fn set_schema(&self, schema: serde_json::Value) { method schema (line 336) | pub async fn schema(&self) -> Option { method validate_input (line 344) | pub async fn validate_input( method validate_train_input (line 359) | pub async fn validate_train_input( method healthcheck (line 375) | pub async fn healthcheck( method submit_prediction (line 397) | pub async fn submit_prediction( method prediction_exists (line 438) | pub fn prediction_exists(&self, id: &str) -> bool { method get_prediction_response (line 446) | pub fn get_prediction_response(&self, id: &str) -> Option bool { method remove_prediction (line 620) | pub fn remove_prediction(&self, id: &str) { method trigger_shutdown (line 624) | pub fn trigger_shutdown(&self) { method shutdown_rx (line 628) | pub fn shutdown_rx(&self) -> watch::Receiver { function build_slot_request (line 640) | fn build_slot_request( type MockOrchestrator (line 684) | struct MockOrchestrator { method new (line 691) | fn new() -> Self { method register_count (line 699) | fn register_count(&self) -> usize { method with_idle_ack (line 703) | fn with_idle_ack(mut self) -> Self { method register_prediction (line 711) | async fn register_prediction( method cancel_by_prediction_id (line 729) | async fn cancel_by_prediction_id( method healthcheck (line 736) | async fn healthcheck( method shutdown (line 742) | async fn shutdown(&self) -> Result<(), crate::orchestrator::Orchestrator... function create_test_pool (line 747) | async fn create_test_pool(num_slots: usize) -> Arc { function create_test_pool_with_slots (line 771) | async fn create_test_pool_with_slots(num_slots: usize) -> (Arc (Arc, SlotId) { function service_new_no_pool_works (line 814) | async fn service_new_no_pool_works() { function service_no_pool_initially (line 825) | async fn service_no_pool_initially() { function shutdown_signal_works (line 833) | async fn shutdown_signal_works() { function submit_fails_when_not_ready (line 846) | async fn submit_fails_when_not_ready() { function cannot_set_ready_without_orchestrator (line 856) | async fn cannot_set_ready_without_orchestrator() { function set_orchestrator_enables_ready_health (line 869) | async fn set_orchestrator_enables_ready_health() { function submit_prediction_succeeds_when_ready (line 886) | async fn submit_prediction_succeeds_when_ready() { function submit_returns_at_capacity_when_no_slots (line 904) | async fn submit_returns_at_capacity_when_no_slots() { function predict_calls_orchestrator_register (line 926) | async fn predict_calls_orchestrator_register() { function health_shows_busy_when_all_slots_used (line 958) | async fn health_shows_busy_when_all_slots_used() { function predict_idle_channel_closed_poison_slot_async (line 982) | async fn predict_idle_channel_closed_poison_slot_async() { function predict_idle_ack_returns_capacity_async (line 1022) | async fn predict_idle_ack_returns_capacity_async() { function predict_send_failure_poison_slot (line 1061) | async fn predict_send_failure_poison_slot() { function cancel_prediction_works (line 1091) | async fn cancel_prediction_works() { function cancel_nonexistent_returns_false (line 1111) | async fn cancel_nonexistent_returns_false() { function sync_guard_cancels_on_drop (line 1117) | async fn sync_guard_cancels_on_drop() { function sync_guard_disarm_prevents_cancel (line 1141) | async fn sync_guard_disarm_prevents_cancel() { function remove_prediction_cleans_up (line 1165) | async fn remove_prediction_cleans_up() { function build_slot_request_small_input_inline (line 1184) | fn build_slot_request_small_input_inline() { function build_slot_request_large_input_spills (line 1213) | fn build_slot_request_large_input_spills() { function build_slot_request_roundtrip (line 1249) | fn build_slot_request_roundtrip() { FILE: crates/coglet/src/setup_log_accumulator.rs type SetupLogAccumulator (line 16) | pub struct SetupLogAccumulator { method new (line 21) | pub fn new(tx: mpsc::UnboundedSender) -> Self { method on_event (line 30) | fn on_event(&self, event: &tracing::Event<'_>, _ctx: Context<'_, S>) { type MessageVisitor (line 47) | struct MessageVisitor { method record_debug (line 52) | fn record_debug(&mut self, field: &tracing::field::Field, value: &dyn ... method record_str (line 61) | fn record_str(&mut self, field: &tracing::field::Field, value: &str) { function drain_accumulated_logs (line 68) | pub fn drain_accumulated_logs(rx: &mut mpsc::UnboundedReceiver) ... FILE: crates/coglet/src/transport/http/routes.rs type HealthCheckResponse (line 23) | pub struct HealthCheckResponse { method from_snapshot (line 33) | pub fn from_snapshot(snapshot: HealthSnapshot, user_healthcheck_error:... type PredictionRequest (line 53) | pub struct PredictionRequest { function default_empty_input (line 68) | fn default_empty_input() -> serde_json::Value { function deserialize_input (line 72) | fn deserialize_input<'de, D>(deserializer: D) -> Result Vec { function generate_prediction_id (line 93) | fn generate_prediction_id() -> String { function root (line 109) | async fn root(State(service): State>) -> Json>) -> ... function write_readiness_file (line 182) | fn write_readiness_file() { function should_respond_async (line 204) | fn should_respond_async(headers: &HeaderMap) -> bool { function extract_trace_context (line 212) | fn extract_trace_context(headers: &HeaderMap) -> TraceContext { function create_prediction (line 225) | async fn create_prediction( function create_prediction_idempotent (line 254) | async fn create_prediction_idempotent( function build_webhook_sender (line 304) | fn build_webhook_sender( function create_prediction_with_id (line 329) | async fn create_prediction_with_id( function cancel_prediction (line 528) | async fn cancel_prediction( function shutdown (line 541) | async fn shutdown(State(service): State>) -> impl... function openapi_schema (line 547) | async fn openapi_schema(State(service): State>) -... function create_training (line 562) | async fn create_training( function create_training_idempotent (line 591) | async fn create_training_idempotent( function cancel_training (line 641) | async fn cancel_training( constant MAX_HTTP_BODY_SIZE (line 653) | const MAX_HTTP_BODY_SIZE: usize = 100 * 1024 * 1024; function routes (line 655) | pub fn routes(service: Arc) -> Router { function response_json (line 679) | async fn response_json(response: axum::response::Response) -> serde_json... function health_check_returns_status_and_version (line 686) | async fn health_check_returns_status_and_version() { function health_check_unknown_when_no_predictor (line 703) | async fn health_check_unknown_when_no_predictor() { function predictions_returns_503_when_not_ready (line 717) | async fn predictions_returns_503_when_not_ready() { function openapi_returns_503_when_schema_not_available (line 744) | async fn openapi_returns_503_when_schema_not_available() { function openapi_returns_schema_when_available (line 760) | async fn openapi_returns_schema_when_available() { type MockOrchestrator (line 792) | struct MockOrchestrator { method new (line 798) | fn new() -> Self { method never_complete (line 806) | fn never_complete() -> Self { method register_prediction (line 816) | async fn register_prediction( method cancel_by_prediction_id (line 829) | async fn cancel_by_prediction_id( method healthcheck (line 836) | async fn healthcheck( method shutdown (line 843) | async fn shutdown(&self) -> Result<(), crate::orchestrator::Orchestrator... function create_test_pool (line 848) | async fn create_test_pool(num_slots: usize) -> Arc { function create_ready_service (line 872) | async fn create_ready_service() -> Arc { function health_check_ready_with_orchestrator (line 882) | async fn health_check_ready_with_orchestrator() { function prediction_sync_success (line 897) | async fn prediction_sync_success() { function prediction_async_returns_accepted (line 919) | async fn prediction_async_returns_accepted() { function prediction_with_custom_id (line 940) | async fn prediction_with_custom_id() { function prediction_idempotent_put (line 961) | async fn prediction_idempotent_put() { function prediction_idempotent_id_mismatch (line 982) | async fn prediction_idempotent_id_mismatch() { function prediction_at_capacity (line 1007) | async fn prediction_at_capacity() { function health_check_busy_when_at_capacity (line 1049) | async fn health_check_busy_when_at_capacity() { function training_routes_work (line 1084) | async fn training_routes_work() { function training_idempotent_put (line 1104) | async fn training_idempotent_put() { function training_idempotent_id_mismatch (line 1125) | async fn training_idempotent_id_mismatch() { function shutdown_triggers_service_shutdown (line 1150) | async fn shutdown_triggers_service_shutdown() { function root_returns_discovery_document (line 1168) | async fn root_returns_discovery_document() { function root_includes_training_urls_when_schema_has_training (line 1206) | async fn root_includes_training_urls_when_schema_has_training() { function root_cog_version_prefers_python_sdk (line 1247) | async fn root_cog_version_prefers_python_sdk() { FILE: crates/coglet/src/transport/http/server.rs type ServerConfig (line 15) | pub struct ServerConfig { method default (line 24) | fn default() -> Self { function serve (line 34) | pub async fn serve(config: ServerConfig, service: Arc... function shutdown_signal (line 66) | async fn shutdown_signal(await_explicit_shutdown: bool, mut shutdown_rx:... function server_config_default (line 118) | fn server_config_default() { FILE: crates/coglet/src/version.rs constant COGLET_VERSION (line 4) | pub const COGLET_VERSION: &str = env!("CARGO_PKG_VERSION"); type VersionInfo (line 8) | pub struct VersionInfo { method new (line 39) | pub fn new() -> Self { method with_git_sha (line 44) | pub fn with_git_sha(mut self, sha: String) -> Self { method with_build_time (line 50) | pub fn with_build_time(mut self, time: String) -> Self { method with_python_sdk (line 56) | pub fn with_python_sdk(mut self, version: String) -> Self { method with_python (line 62) | pub fn with_python(mut self, version: String) -> Self { method default (line 26) | fn default() -> Self { function version_info_has_coglet_version (line 73) | fn version_info_has_coglet_version() { function version_info_builder_pattern (line 81) | fn version_info_builder_pattern() { function version_info_serializes_minimal (line 95) | fn version_info_serializes_minimal() { function version_info_serializes_full (line 108) | fn version_info_serializes_full() { FILE: crates/coglet/src/webhook.rs type WebhookEventType (line 28) | pub enum WebhookEventType { method is_terminal (line 37) | pub fn is_terminal(&self) -> bool { method all (line 41) | pub fn all() -> HashSet { type WebhookSenderError (line 50) | pub enum WebhookSenderError { type WebhookConfig (line 56) | pub struct WebhookConfig { method default (line 65) | fn default() -> Self { type TraceContext (line 84) | pub struct TraceContext { type WebhookSender (line 89) | pub struct WebhookSender { method new (line 98) | pub fn new(url: String, config: WebhookConfig) -> Result &str { method should_send (line 139) | fn should_send(&self, event: WebhookEventType) -> bool { method update_last_sent (line 162) | fn update_last_sent(&self) { method build_request (line 168) | fn build_request(&self, payload: &serde_json::Value) -> reqwest::Reque... method send (line 182) | pub fn send(&self, event: WebhookEventType, payload: &serde_json::Valu... method send_terminal (line 198) | pub async fn send_terminal(&self, event: WebhookEventType, payload: &s... method send_terminal_sync (line 268) | pub fn send_terminal_sync(&self, payload: &serde_json::Value) { function config_defaults (line 381) | fn config_defaults() { function event_is_terminal (line 390) | fn event_is_terminal() { function test_config (line 397) | fn test_config() -> WebhookConfig { function send_terminal_posts_json (line 407) | async fn send_terminal_posts_json() { function send_terminal_retries_on_500 (line 429) | async fn send_terminal_retries_on_500() { function send_terminal_no_retry_on_400 (line 458) | async fn send_terminal_no_retry_on_400() { function send_terminal_respects_filter (line 480) | async fn send_terminal_respects_filter() { function send_non_terminal_fires_and_forgets (line 506) | async fn send_non_terminal_fires_and_forgets() { function send_non_terminal_logs_throttled (line 528) | async fn send_non_terminal_logs_throttled() { function send_output_not_throttled (line 559) | async fn send_output_not_throttled() { function send_terminal_sync_posts_json (line 590) | async fn send_terminal_sync_posts_json() { function send_terminal_sync_retries_on_500 (line 607) | async fn send_terminal_sync_retries_on_500() { FILE: crates/coglet/src/worker.rs function increment_dropped_log_count (line 35) | pub fn increment_dropped_log_count() { function report_dropped_logs (line 41) | fn report_dropped_logs(tx: &mpsc::Sender, interval_mill... type FatalContext (line 55) | struct FatalContext { function init_fatal_context (line 61) | fn init_fatal_context(tx: mpsc::Sender) { function install_panic_hook (line 71) | fn install_panic_hook() { function init_worker_tracing (line 103) | fn init_worker_tracing(tx: mpsc::Sender) { type SlotWriter (line 142) | type SlotWriter = type SlotSender (line 150) | pub struct SlotSender { method new (line 157) | pub fn new(tx: mpsc::UnboundedSender, output_dir: PathBu... method next_output_path (line 166) | fn next_output_path(&self, extension: &str) -> PathBuf { method send_log (line 171) | pub fn send_log(&self, source: LogSource, data: &str) -> io::Result<()> { method write_file_output (line 190) | pub fn write_file_output( method send_file_output (line 205) | pub fn send_file_output(&self, path: PathBuf, mime_type: Option io::Result<()> { function build_output_message (line 243) | fn build_output_message( type SetupError (line 272) | pub enum SetupError { method load (line 287) | pub fn load(message: impl Into) -> Self { method setup (line 293) | pub fn setup(message: impl Into) -> Self { method internal (line 299) | pub fn internal(message: impl Into) -> Self { type PredictHandler (line 308) | pub trait PredictHandler: Send + Sync + 'static { method setup (line 310) | async fn setup(&self) -> Result<(), SetupError>; method predict (line 313) | async fn predict( method cancel (line 323) | fn cancel(&self, slot: SlotId); method healthcheck (line 326) | async fn healthcheck(&self) -> HealthcheckResult { constant BUNDLED_SCHEMA_PATH (line 333) | const BUNDLED_SCHEMA_PATH: &str = ".cog/openapi_schema.json"; function load_bundled_schema (line 340) | fn load_bundled_schema() -> Option { type PredictionOutcome (line 370) | pub enum PredictionOutcome { type PredictResult (line 385) | pub struct PredictResult { method success (line 390) | pub fn success(output: serde_json::Value, predict_time: f64, is_stream... method failed (line 400) | pub fn failed(error: String, predict_time: f64) -> Self { method cancelled (line 409) | pub fn cancelled(predict_time: f64) -> Self { type SetupLogHook (line 420) | pub type SetupLogHook = type WorkerConfig (line 423) | pub struct WorkerConfig { method default (line 430) | fn default() -> Self { type SlotCompletion (line 438) | struct SlotCompletion { method idle (line 443) | fn idle(slot: SlotId) -> Self { method poisoned (line 449) | fn poisoned(slot: SlotId, error: impl Into) -> Self { function run_worker (line 460) | pub async fn run_worker( function slot_reader_task (line 786) | async fn slot_reader_task( function run_prediction (line 810) | async fn run_prediction( function predict_result_success (line 919) | fn predict_result_success() { function predict_result_success_stream (line 925) | fn predict_result_success_stream() { function predict_result_failed (line 937) | fn predict_result_failed() { function predict_result_cancelled (line 946) | fn predict_result_cancelled() { function worker_config_default (line 952) | fn worker_config_default() { FILE: crates/coglet/src/worker_tracing_layer.rs type WorkerTracingLayer (line 15) | pub struct WorkerTracingLayer { method new (line 21) | pub fn new(tx: mpsc::Sender) -> Self { method level_to_string (line 33) | fn level_to_string(level: &Level) -> &'static str { method on_event (line 48) | fn on_event(&self, event: &tracing::Event<'_>, _ctx: Context<'_, S>) { type MessageVisitor (line 93) | struct MessageVisitor { method record_debug (line 98) | fn record_debug(&mut self, field: &tracing::field::Field, value: &dyn ... method record_str (line 107) | fn record_str(&mut self, field: &tracing::field::Field, value: &str) { FILE: integration-tests/concurrent/concurrent_test.go function TestConcurrentPredictions (line 36) | func TestConcurrentPredictions(t *testing.T) { type predictionResult (line 142) | type predictionResult struct function makePrediction (line 148) | func makePrediction(serverURL string, idx int) predictionResult { function waitForServerReady (line 175) | func waitForServerReady(serverURL string, timeout time.Duration) bool { function waitForServerStatus (line 212) | func waitForServerStatus(serverURL string, targetStatus string, timeout ... function allocatePort (line 247) | func allocatePort() (int, error) { constant cogYAML (line 258) | cogYAML = `build: constant predictPy (line 265) | predictPy = `import asyncio function TestConcurrentAboveLimit (line 277) | func TestConcurrentAboveLimit(t *testing.T) { constant aboveLimitCogYAML (line 373) | aboveLimitCogYAML = `build: function TestSIGTERMDuringSetup (line 381) | func TestSIGTERMDuringSetup(t *testing.T) { FILE: integration-tests/harness/cmd_pty.go type PtyRunCommand (line 17) | type PtyRunCommand struct method Name (line 21) | func (c *PtyRunCommand) Name() string { return "pty-run" } method Run (line 37) | func (c *PtyRunCommand) Run(ts *testscript.TestScript, neg bool, args ... FILE: integration-tests/harness/command.go type Command (line 6) | type Command interface type CommandFunc (line 17) | type CommandFunc struct method Name (line 22) | func (c CommandFunc) Name() string { return c.name } method Run (line 23) | func (c CommandFunc) Run(ts *testscript.TestScript, neg bool, args []s... function NewCommand (line 28) | func NewCommand(name string, fn func(ts *testscript.TestScript, neg bool... FILE: integration-tests/harness/harness.go type serverInfo (line 43) | type serverInfo struct type registryInfo (line 49) | type registryInfo struct type mockUploadRecord (line 56) | type mockUploadRecord struct type mockUploadServer (line 64) | type mockUploadServer struct type webhookResult (line 72) | type webhookResult struct type webhookServer (line 83) | type webhookServer struct type Harness (line 91) | type Harness struct method Commands (line 240) | func (h *Harness) Commands() map[string]func(ts *testscript.TestScript... method cmdCog (line 278) | func (h *Harness) cmdCog(ts *testscript.TestScript, neg bool, args []s... method Setup (line 307) | func (h *Harness) Setup(env *testscript.Env) error { method cmdCogServe (line 395) | func (h *Harness) cmdCogServe(ts *testscript.TestScript, neg bool, arg... method cmdCurl (line 483) | func (h *Harness) cmdCurl(ts *testscript.TestScript, neg bool, args []... method StopServer (line 616) | func (h *Harness) StopServer(ts *testscript.TestScript) { method stopServerByWorkDir (line 622) | func (h *Harness) stopServerByWorkDir(workDir string) { method cmdWaitFor (line 729) | func (h *Harness) cmdWaitFor(ts *testscript.TestScript, neg bool, args... method cmdDockerRun (line 819) | func (h *Harness) cmdDockerRun(ts *testscript.TestScript, neg bool, ar... method cmdRegistryStart (line 866) | func (h *Harness) cmdRegistryStart(ts *testscript.TestScript, neg bool... method stopRegistryByWorkDir (line 905) | func (h *Harness) stopRegistryByWorkDir(workDir string) { method cmdRegistrySeed (line 930) | func (h *Harness) cmdRegistrySeed(ts *testscript.TestScript, neg bool,... method cmdRegistryInspect (line 978) | func (h *Harness) cmdRegistryInspect(ts *testscript.TestScript, neg bo... method cmdDockerPush (line 1013) | func (h *Harness) cmdDockerPush(ts *testscript.TestScript, neg bool, a... method cmdMockWeights (line 1093) | func (h *Harness) cmdMockWeights(ts *testscript.TestScript, neg bool, ... method cmdUploadServerStart (line 1244) | func (h *Harness) cmdUploadServerStart(ts *testscript.TestScript, neg ... method cmdUploadServerCount (line 1307) | func (h *Harness) cmdUploadServerCount(ts *testscript.TestScript, neg ... method stopUploadServerByWorkDir (line 1343) | func (h *Harness) stopUploadServerByWorkDir(workDir string) { method cmdWebhookServerStart (line 1367) | func (h *Harness) cmdWebhookServerStart(ts *testscript.TestScript, neg... method cmdWebhookServerWait (line 1465) | func (h *Harness) cmdWebhookServerWait(ts *testscript.TestScript, neg ... method stopWebhookServerByWorkDir (line 1502) | func (h *Harness) stopWebhookServerByWorkDir(workDir string) { function New (line 112) | func New() (*Harness, error) { function ResolveCogBinary (line 136) | func ResolveCogBinary() (string, error) { function buildCogBinary (line 156) | func buildCogBinary() (string, error) { function findRepoRoot (line 203) | func findRepoRoot() (string, error) { function runCommand (line 231) | func runCommand(dir string, name string, args ...string) error { function generateUniqueImageName (line 364) | func generateUniqueImageName() string { function removeDockerImage (line 375) | func removeDockerImage(imageName string) { function allocatePort (line 658) | func allocatePort() (int, error) { type healthCheckResponse (line 668) | type healthCheckResponse struct function waitForServer (line 675) | func waitForServer(serverURL string, timeout time.Duration) bool { function isAbsoluteImageRef (line 967) | func isAbsoluteImageRef(ref string) bool { type mockWeightsLock (line 1066) | type mockWeightsLock struct type mockWeightFile (line 1074) | type mockWeightFile struct function parseSize (line 1200) | func parseSize(s string) (int64, error) { FILE: integration-tests/login/login_test.go function TestLoginGenericRegistryPTY (line 38) | func TestLoginGenericRegistryPTY(t *testing.T) { function TestLoginProviderRouting (line 232) | func TestLoginProviderRouting(t *testing.T) { function TestLoginEnvironmentVariable (line 335) | func TestLoginEnvironmentVariable(t *testing.T) { function TestLoginHelp (line 434) | func TestLoginHelp(t *testing.T) { function TestLoginSuggestFor (line 467) | func TestLoginSuggestFor(t *testing.T) { FILE: integration-tests/suite_test.go function TestMain (line 24) | func TestMain(m *testing.M) { function TestIntegration (line 37) | func TestIntegration(t *testing.T) { function condition (line 68) | func condition(cond string) (bool, error) { FILE: noxfile.py function _find_compatible_wheel (line 23) | def _find_compatible_wheel(pattern: str) -> str | None: function _install_coglet (line 57) | def _install_coglet(session: nox.Session) -> None: function _install_package (line 68) | def _install_package(session: nox.Session) -> None: function tests (line 82) | def tests(session: nox.Session) -> None: function typecheck (line 97) | def typecheck(session: nox.Session) -> None: function coglet_tests (line 105) | def coglet_tests(session: nox.Session) -> None: FILE: pkg/cli/baseimage.go function NewBaseImageRootCommand (line 29) | func NewBaseImageRootCommand() (*cobra.Command, error) { function newBaseImageGenerateMatrix (line 57) | func newBaseImageGenerateMatrix() *cobra.Command { function newBaseImageDockerfileCommand (line 128) | func newBaseImageDockerfileCommand() *cobra.Command { function newBaseImageBuildCommand (line 155) | func newBaseImageBuildCommand() *cobra.Command { function addBaseImageFlags (line 204) | func addBaseImageFlags(cmd *cobra.Command) { function baseImageGeneratorFromFlags (line 211) | func baseImageGeneratorFromFlags(ctx context.Context) (*dockerfile.BaseI... FILE: pkg/cli/build.go constant useCogBaseImageFlagKey (line 31) | useCogBaseImageFlagKey = "use-cog-base-image" function newBuildCommand (line 33) | func newBuildCommand() *cobra.Command { function buildCommand (line 73) | func buildCommand(cmd *cobra.Command, args []string) error { function addBuildProgressOutputFlag (line 109) | func addBuildProgressOutputFlag(cmd *cobra.Command) { function addSecretsFlag (line 120) | func addSecretsFlag(cmd *cobra.Command) { function addNoCacheFlag (line 124) | func addNoCacheFlag(cmd *cobra.Command) { function addSeparateWeightsFlag (line 128) | func addSeparateWeightsFlag(cmd *cobra.Command) { function addSchemaFlag (line 132) | func addSchemaFlag(cmd *cobra.Command) { function addUseCudaBaseImageFlag (line 136) | func addUseCudaBaseImageFlag(cmd *cobra.Command) { function addDockerfileFlag (line 140) | func addDockerfileFlag(cmd *cobra.Command) { function addUseCogBaseImageFlag (line 149) | func addUseCogBaseImageFlag(cmd *cobra.Command) { function addBuildTimestampFlag (line 153) | func addBuildTimestampFlag(cmd *cobra.Command) { function addStripFlag (line 158) | func addStripFlag(cmd *cobra.Command) { function addPrecompileFlag (line 164) | func addPrecompileFlag(cmd *cobra.Command) { function addConfigFlag (line 170) | func addConfigFlag(cmd *cobra.Command) { function checkMutuallyExclusiveFlags (line 174) | func checkMutuallyExclusiveFlags(cmd *cobra.Command, args []string) error { function DetermineUseCogBaseImage (line 188) | func DetermineUseCogBaseImage(cmd *cobra.Command) *bool { function buildOptionsFromFlags (line 199) | func buildOptionsFromFlags(cmd *cobra.Command, imageName string, annotat... FILE: pkg/cli/debug.go function newDebugCommand (line 19) | func newDebugCommand() *cobra.Command { function cmdDockerfile (line 38) | func cmdDockerfile(cmd *cobra.Command, args []string) error { FILE: pkg/cli/init-templates/base/predict.py class Predictor (line 7) | class Predictor(BasePredictor): method setup (line 8) | def setup(self) -> None: method predict (line 12) | def predict( FILE: pkg/cli/init.go function newInitCommand (line 21) | func newInitCommand() *cobra.Command { function initCommand (line 39) | func initCommand(cmd *cobra.Command, args []string) error { function processTemplateDirectory (line 77) | func processTemplateDirectory(fs embed.FS, templateDir, subDir, cwd stri... function processTemplateFile (line 103) | func processTemplateFile(fs embed.FS, templateDir, filename, cwd string)... function downloadAgentsFile (line 154) | func downloadAgentsFile() ([]byte, error) { FILE: pkg/cli/init_test.go function TestInit (line 11) | func TestInit(t *testing.T) { function TestInitSkipExisting (line 25) | func TestInitSkipExisting(t *testing.T) { FILE: pkg/cli/inspect.go type InspectOutput (line 18) | type InspectOutput struct type InspectIndex (line 27) | type InspectIndex struct type InspectManifest (line 35) | type InspectManifest struct type InspectLayer (line 48) | type InspectLayer struct function newInspectCommand (line 54) | func newInspectCommand() *cobra.Command { function inspectCommand (line 80) | func inspectCommand(cmd *cobra.Command, args []string, localOnly, remote... function buildInspectOutput (line 129) | func buildInspectOutput(ctx context.Context, reference string, m *model.... function buildManifestEntry (line 186) | func buildManifestEntry(im model.IndexManifest) InspectManifest { function fetchLayers (line 213) | func fetchLayers(ctx context.Context, reference, digest string, reg regi... type rawStep (line 243) | type rawStep struct function streamRaw (line 249) | func streamRaw(ctx context.Context, reference string, m *model.Model, re... function printInspectText (line 322) | func printInspectText(out *InspectOutput) { function printManifestText (line 362) | func printManifestText(m InspectManifest, indent string) { FILE: pkg/cli/login.go function newLoginCommand (line 12) | func newLoginCommand() *cobra.Command { function login (line 33) | func login(cmd *cobra.Command, args []string) error { FILE: pkg/cli/predict.go constant StdinPath (line 35) | StdinPath = "-" function newPredictCommand (line 46) | func newPredictCommand() *cobra.Command { function readStdin (line 96) | func readStdin() (string, error) { function parseJSONInput (line 105) | func parseJSONInput(jsonInput string) (map[string]any, error) { function transformPathsToBase64URLs (line 146) | func transformPathsToBase64URLs(inputs map[string]any) (map[string]any, ... function cmdPredict (line 179) | func cmdPredict(cmd *cobra.Command, args []string) error { function isURI (line 316) | func isURI(ref *openapi3.Schema) bool { function predictJSONInputs (line 320) | func predictJSONInputs(predictor predict.Predictor, jsonInput string, ou... function predictIndividualInputs (line 350) | func predictIndividualInputs(predictor predict.Predictor, inputFlags []s... function runPrediction (line 364) | func runPrediction(predictor predict.Predictor, inputs predict.Inputs, o... function ensureOutputWriteable (line 532) | func ensureOutputWriteable(outputPath string, fallbackPath string) (stri... function prettyJSONMarshal (line 577) | func prettyJSONMarshal(v any) ([]byte, error) { function processFileOutputs (line 589) | func processFileOutputs(output any, schema *openapi3.Schema, destination... function parseInputFlags (line 628) | func parseInputFlags(inputs []string, schema *openapi3.T, isTrain ...boo... function addSetupTimeoutFlag (line 654) | func addSetupTimeoutFlag(cmd *cobra.Command) { FILE: pkg/cli/predict_test.go function TestExtractOutputSchemaFromMalformedSchema (line 10) | func TestExtractOutputSchemaFromMalformedSchema(t *testing.T) { function safeExtractOutputSchema (line 111) | func safeExtractOutputSchema(schema *openapi3.T, url string) *openapi3.S... function TestExtractOutputSchemaFromValidSchema (line 137) | func TestExtractOutputSchemaFromValidSchema(t *testing.T) { FILE: pkg/cli/push.go function newPushCommand (line 18) | func newPushCommand() *cobra.Command { function push (line 52) | func push(cmd *cobra.Command, args []string) error { FILE: pkg/cli/root.go function NewRootCommand (line 14) | func NewRootCommand() (*cobra.Command, error) { function setPersistentFlags (line 62) | func setPersistentFlags(cmd *cobra.Command) { FILE: pkg/cli/run.go function addGpusFlag (line 22) | func addGpusFlag(cmd *cobra.Command) { function newRunCommand (line 26) | func newRunCommand() *cobra.Command { function run (line 69) | func run(cmd *cobra.Command, args []string) error { FILE: pkg/cli/serve.go function newServeCommand (line 22) | func newServeCommand() *cobra.Command { function serveBuildOptions (line 62) | func serveBuildOptions(cmd *cobra.Command) model.BuildOptions { function cmdServe (line 72) | func cmdServe(cmd *cobra.Command, arg []string) error { FILE: pkg/cli/train.go function newTrainCommand (line 26) | func newTrainCommand() *cobra.Command { function cmdTrain (line 56) | func cmdTrain(cmd *cobra.Command, args []string) error { FILE: pkg/cli/train_test.go function TestTrainCommandIsDeprecated (line 9) | func TestTrainCommandIsDeprecated(t *testing.T) { FILE: pkg/cli/weights.go function newWeightsCommand (line 19) | func newWeightsCommand() *cobra.Command { function newWeightsBuildCommand (line 33) | func newWeightsBuildCommand() *cobra.Command { function weightsBuildCommand (line 47) | func weightsBuildCommand(cmd *cobra.Command, args []string) error { function formatSize (line 95) | func formatSize(bytes int64) string { function newWeightsPushCommand (line 114) | func newWeightsPushCommand() *cobra.Command { function weightsPushCommand (line 131) | func weightsPushCommand(cmd *cobra.Command, args []string) error { FILE: pkg/cli/weights_inspect.go type localWeight (line 18) | type localWeight struct type WeightsInspectOutput (line 25) | type WeightsInspectOutput struct type WeightInspectEntry (line 31) | type WeightInspectEntry struct type WeightLocalState (line 39) | type WeightLocalState struct type WeightRemoteLayer (line 47) | type WeightRemoteLayer struct type WeightRemoteState (line 54) | type WeightRemoteState struct function newWeightsInspectCommand (line 64) | func newWeightsInspectCommand() *cobra.Command { function weightsInspectCommand (line 82) | func weightsInspectCommand(cmd *cobra.Command, args []string, jsonOutput... function resolveWeightsByTag (line 204) | func resolveWeightsByTag(ctx context.Context, repo string, localWeights ... function printWeightsInspectText (line 260) | func printWeightsInspectText(out *WeightsInspectOutput) { function fileExists (line 303) | func fileExists(path string) bool { FILE: pkg/config/build_options.go type BuildOptions (line 7) | type BuildOptions struct function DefaultBuildOptions (line 19) | func DefaultBuildOptions() BuildOptions { FILE: pkg/config/compatibility.go type TFCompatibility (line 26) | type TFCompatibility struct method UnmarshalJSON (line 35) | func (compat *TFCompatibility) UnmarshalJSON(data []byte) error { type TorchCompatibility (line 55) | type TorchCompatibility struct method TorchVersion (line 65) | func (c *TorchCompatibility) TorchVersion() string { method TorchvisionVersion (line 69) | func (c *TorchCompatibility) TorchvisionVersion() string { type CUDABaseImage (line 73) | type CUDABaseImage struct method ImageTag (line 81) | func (i *CUDABaseImage) ImageTag() string { function init (line 97) | func init() { function cudaVersionFromTorchPlusVersion (line 122) | func cudaVersionFromTorchPlusVersion(ver string) (string, string) { function cudasFromTorch (line 154) | func cudasFromTorch(ver string) ([]string, error) { function cudaFromTF (line 186) | func cudaFromTF(ver string) (cuda string, cuDNN string, err error) { function compatibleCuDNNsForCUDA (line 195) | func compatibleCuDNNsForCUDA(cuda string) []string { function defaultCUDA (line 205) | func defaultCUDA() string { function latestCUDAFrom (line 210) | func latestCUDAFrom(cudas []string) string { function latestCuDNNForCUDA (line 229) | func latestCuDNNForCUDA(cuda string) (string, error) { function versionGreater (line 246) | func versionGreater(a string, b string) (bool, error) { function cudaBaseImageFor (line 259) | func cudaBaseImageFor(cuda string, cuDNN string) (string, error) { function tfGPUPackage (line 280) | func tfGPUPackage(ver string, cuda string) (name string, cpuVersion stri... function torchCPUPackage (line 291) | func torchCPUPackage(ver, goos, goarch string) (name, cpuVersion, findLi... function torchGPUPackage (line 302) | func torchGPUPackage(ver string, cuda string) (name, cpuVersion, findLin... function torchvisionCPUPackage (line 339) | func torchvisionCPUPackage(ver, goos, goarch string) (name, cpuVersion, ... function torchvisionGPUPackage (line 349) | func torchvisionGPUPackage(ver, cuda string) (name, cpuVersion, findLink... function torchStripCPUSuffixForM1 (line 389) | func torchStripCPUSuffixForM1(version string, goos string, goarch string... FILE: pkg/config/compatibility_test.go function TestLatestCuDNNForCUDA (line 9) | func TestLatestCuDNNForCUDA(t *testing.T) { function TestCudasFromTorchWithCUVersionModifier (line 15) | func TestCudasFromTorchWithCUVersionModifier(t *testing.T) { FILE: pkg/config/config.go constant MinimumMajorPythonVersion (line 31) | MinimumMajorPythonVersion int = 3 constant MinimumMinorPythonVersion (line 32) | MinimumMinorPythonVersion int = 10 constant MinimumMinorPythonVersionForConcurrency (line 33) | MinimumMinorPythonVersionForConcurrency int = 11 constant MinimumMajorCudaVersion (line 34) | MinimumMajorCudaVersion int = 11 constant DefaultPythonVersion (line 35) | DefaultPythonVersion string = "3.13" type RunItem (line 38) | type RunItem struct method UnmarshalYAML (line 97) | func (r *RunItem) UnmarshalYAML(unmarshal func(any) error) error { method UnmarshalJSON (line 135) | func (r *RunItem) UnmarshalJSON(data []byte) error { type Build (line 47) | type Build struct type Concurrency (line 65) | type Concurrency struct type WeightSource (line 70) | type WeightSource struct type Config (line 76) | type Config struct method CUDABaseImageTag (line 171) | func (c *Config) CUDABaseImageTag() (string, error) { method TorchVersion (line 175) | func (c *Config) TorchVersion() (string, bool) { method TorchvisionVersion (line 179) | func (c *Config) TorchvisionVersion() (string, bool) { method TorchaudioVersion (line 183) | func (c *Config) TorchaudioVersion() (string, bool) { method TensorFlowVersion (line 187) | func (c *Config) TensorFlowVersion() (string, bool) { method cudasFromTorch (line 191) | func (c *Config) cudasFromTorch() (torchVersion string, torchCUDAs []s... method cudaFromTF (line 202) | func (c *Config) cudaFromTF() (tfVersion string, tfCUDA string, tfCuDN... method pythonPackageVersion (line 213) | func (c *Config) pythonPackageVersion(name string) (version string, ok... method Complete (line 248) | func (c *Config) Complete(projectDir string) error { method PythonRequirementsForArch (line 286) | func (c *Config) PythonRequirementsForArch(goos string, goarch string,... method pythonPackageForArch (line 352) | func (c *Config) pythonPackageForArch(pkg, goos, goarch string) (actua... method validateAndCompleteCUDA (line 428) | func (c *Config) validateAndCompleteCUDA() error { method RequirementsFile (line 525) | func (c *Config) RequirementsFile(projectDir string) string { method ParsedEnvironment (line 529) | func (c *Config) ParsedEnvironment() map[string]string { method loadEnvironment (line 533) | func (c *Config) loadEnvironment() error { function defaultConfig (line 88) | func defaultConfig() *Config { function splitPythonVersion (line 227) | func splitPythonVersion(version string) (major int, minor int, err error) { function validateCudaVersion (line 411) | func validateCudaVersion(cudaVersion string) error { FILE: pkg/config/config_file.go type configFile (line 14) | type configFile struct type buildFile (line 25) | type buildFile struct method GetGPU (line 162) | func (b *buildFile) GetGPU() bool { type runItemFile (line 41) | type runItemFile struct method UnmarshalYAML (line 67) | func (r *runItemFile) UnmarshalYAML(unmarshal func(any) error) error { method UnmarshalJSON (line 115) | func (r *runItemFile) UnmarshalJSON(data []byte) error { type mountFile (line 47) | type mountFile struct type weightFile (line 54) | type weightFile struct type concurrencyFile (line 61) | type concurrencyFile struct FILE: pkg/config/config_test.go function TestValidateCudaVersion (line 15) | func TestValidateCudaVersion(t *testing.T) { function assertMinorVersion (line 65) | func assertMinorVersion(t *testing.T, expected, actual string) { function TestPythonPackagesAndRequirementsCantBeUsedTogether (line 83) | func TestPythonPackagesAndRequirementsCantBeUsedTogether(t *testing.T) { function TestPythonRequirementsResolvesPythonPackagesAndCudaVersions (line 98) | func TestPythonRequirementsResolvesPythonPackagesAndCudaVersions(t *test... function TestPythonRequirementsResolvesPythonPackagesAndCudaVersionsWithExtraIndexURL (line 128) | func TestPythonRequirementsResolvesPythonPackagesAndCudaVersionsWithExtr... function TestPythonRequirementsWorksWithLinesCogCannotParse (line 158) | func TestPythonRequirementsWorksWithLinesCogCannotParse(t *testing.T) { function TestValidateAndCompleteCUDAForAllTF (line 191) | func TestValidateAndCompleteCUDAForAllTF(t *testing.T) { function TestValidateAndCompleteCUDAForAllTorch (line 210) | func TestValidateAndCompleteCUDAForAllTorch(t *testing.T) { function TestValidateAndCompleteCUDAForSelectedTorch (line 234) | func TestValidateAndCompleteCUDAForSelectedTorch(t *testing.T) { function TestUnsupportedTorch (line 260) | func TestUnsupportedTorch(t *testing.T) { function TestUnsupportedTensorflow (line 296) | func TestUnsupportedTensorflow(t *testing.T) { function TestPythonPackagesForArchTorchGPU (line 333) | func TestPythonPackagesForArchTorchGPU(t *testing.T) { function TestPythonPackagesForArchTorchCPU (line 362) | func TestPythonPackagesForArchTorchCPU(t *testing.T) { function TestPythonPackagesForArchTensorflowGPU (line 389) | func TestPythonPackagesForArchTensorflowGPU(t *testing.T) { function TestPythonPackagesBothTorchAndTensorflow (line 419) | func TestPythonPackagesBothTorchAndTensorflow(t *testing.T) { function TestCUDABaseImageTag (line 444) | func TestCUDABaseImageTag(t *testing.T) { function TestBuildRunItemStringYAML (line 463) | func TestBuildRunItemStringYAML(t *testing.T) { function TestBuildRunItemStringJSON (line 483) | func TestBuildRunItemStringJSON(t *testing.T) { function TestBuildRunItemDictYAML (line 505) | func TestBuildRunItemDictYAML(t *testing.T) { function TestBuildRunItemDictJSON (line 533) | func TestBuildRunItemDictJSON(t *testing.T) { function TestTorchWithExistingExtraIndexURL (line 568) | func TestTorchWithExistingExtraIndexURL(t *testing.T) { function TestBlankBuild (line 590) | func TestBlankBuild(t *testing.T) { function TestPythonRequirementsForArchWithAddedPackage (line 610) | func TestPythonRequirementsForArchWithAddedPackage(t *testing.T) { function TestParseTests (line 634) | func TestParseTests(t *testing.T) { function TestConfigMarshal (line 649) | func TestConfigMarshal(t *testing.T) { function TestAbsolutePathInPythonRequirements (line 660) | func TestAbsolutePathInPythonRequirements(t *testing.T) { function TestWeightsWithNameYAML (line 679) | func TestWeightsWithNameYAML(t *testing.T) { function TestWeightsWithoutNameYAML (line 706) | func TestWeightsWithoutNameYAML(t *testing.T) { function TestWeightsWithNameJSON (line 725) | func TestWeightsWithNameJSON(t *testing.T) { function TestSDKVersionConfig (line 759) | func TestSDKVersionConfig(t *testing.T) { function TestSDKVersionConfigEmpty (line 771) | func TestSDKVersionConfigEmpty(t *testing.T) { function TestSDKVersionConfigPreRelease (line 782) | func TestSDKVersionConfigPreRelease(t *testing.T) { function TestSDKVersionConfigBelowMinimumExplodesInGenerator (line 794) | func TestSDKVersionConfigBelowMinimumExplodesInGenerator(t *testing.T) { FILE: pkg/config/env.go function validateEnvName (line 40) | func validateEnvName(name string) error { function parseAndValidateEnvironment (line 60) | func parseAndValidateEnvironment(input []string) (map[string]string, err... FILE: pkg/config/env_variables_test.go function TestEnvironmentConfig (line 11) | func TestEnvironmentConfig(t *testing.T) { FILE: pkg/config/errors.go type ConfigError (line 10) | type ConfigError interface type ParseError (line 16) | type ParseError struct method Error (line 21) | func (e *ParseError) Error() string { method Unwrap (line 25) | func (e *ParseError) Unwrap() error { method ConfigError (line 29) | func (e *ParseError) ConfigError() {} type SchemaError (line 33) | type SchemaError struct method Error (line 38) | func (e *SchemaError) Error() string { method ConfigError (line 42) | func (e *SchemaError) ConfigError() {} type ValidationError (line 46) | type ValidationError struct method Error (line 52) | func (e *ValidationError) Error() string { method ConfigError (line 59) | func (e *ValidationError) ConfigError() {} type DeprecationWarning (line 63) | type DeprecationWarning struct method Error (line 69) | func (w *DeprecationWarning) Error() string { method ConfigError (line 76) | func (w *DeprecationWarning) ConfigError() {} type CompatibilityError (line 79) | type CompatibilityError struct method Error (line 87) | func (e *CompatibilityError) Error() string { method ConfigError (line 92) | func (e *CompatibilityError) ConfigError() {} type ValidationResult (line 95) | type ValidationResult struct method HasErrors (line 101) | func (r *ValidationResult) HasErrors() bool { method HasWarnings (line 106) | func (r *ValidationResult) HasWarnings() bool { method Err (line 111) | func (r *ValidationResult) Err() error { method AddError (line 119) | func (r *ValidationResult) AddError(err error) { method AddWarning (line 124) | func (r *ValidationResult) AddWarning(w DeprecationWarning) { function NewValidationResult (line 129) | func NewValidationResult() *ValidationResult { FILE: pkg/config/image_name.go function DockerImageName (line 10) | func DockerImageName(projectDir string) string { FILE: pkg/config/image_name_test.go function TestDockerImageName (line 9) | func TestDockerImageName(t *testing.T) { FILE: pkg/config/load.go constant maxSearchDepth (line 13) | maxSearchDepth = 100 type LoadResult (line 16) | type LoadResult struct function Load (line 26) | func Load(r io.Reader, projectDir string) (*LoadResult, error) { function GetProjectDir (line 64) | func GetProjectDir(configFilename string) (string, error) { function findConfigPathInDirectory (line 77) | func findConfigPathInDirectory(dir string, configFilename string) (confi... function findProjectRootDir (line 91) | func findProjectRootDir(startDir string, configFilename string) (string,... FILE: pkg/config/load_test.go constant testConfig (line 11) | testConfig = ` function TestFindProjectRootDirShouldFindParentDir (line 21) | func TestFindProjectRootDirShouldFindParentDir(t *testing.T) { function TestFindProjectRootDirShouldReturnErrIfNoConfig (line 36) | func TestFindProjectRootDirShouldReturnErrIfNoConfig(t *testing.T) { FILE: pkg/config/parse.go function parse (line 17) | func parse(r io.Reader) (*configFile, error) { function parseFile (line 29) | func parseFile(filename string) (*configFile, error) { function parseBytes (line 62) | func parseBytes(contents []byte) (*configFile, error) { function FromYAML (line 86) | func FromYAML(contents []byte) (*Config, error) { function configFileToConfig (line 99) | func configFileToConfig(cfg *configFile) (*Config, error) { FILE: pkg/config/validate.go type ValidateOption (line 23) | type ValidateOption type validateOptions (line 25) | type validateOptions struct function WithProjectDir (line 32) | func WithProjectDir(dir string) ValidateOption { function WithRequirementsFS (line 39) | func WithRequirementsFS(fsys fs.FS) ValidateOption { function WithStrictDeprecations (line 46) | func WithStrictDeprecations() ValidateOption { function ValidateConfigFile (line 55) | func ValidateConfigFile(cfg *configFile, opts ...ValidateOption) *Valida... function validateSchema (line 90) | func validateSchema(cfg *configFile) error { function validatePredict (line 109) | func validatePredict(cfg *configFile, result *ValidationResult) { function validateTrain (line 125) | func validateTrain(cfg *configFile, result *ValidationResult) { function validateBuild (line 141) | func validateBuild(cfg *configFile, opts *validateOptions, result *Valid... function validatePythonVersion (line 189) | func validatePythonVersion(version string) error { function validateCUDAVersion (line 230) | func validateCUDAVersion(cudaVersion string) error { function validateRequirementsFile (line 261) | func validateRequirementsFile(reqPath string, opts *validateOptions) err... function validateGPUConfig (line 292) | func validateGPUConfig(cfg *configFile, opts *validateOptions, result *V... function loadRequirementsForValidation (line 327) | func loadRequirementsForValidation(reqPath string, opts *validateOptions... function parseRequirementsContent (line 349) | func parseRequirementsContent(content string) []string { function validateFrameworkCompatibility (line 363) | func validateFrameworkCompatibility(cfg *configFile, reqs []string, resu... function findPackageVersion (line 396) | func findPackageVersion(reqs []string, name string) string { function validateEnvironment (line 410) | func validateEnvironment(cfg *configFile, result *ValidationResult) { function validateConcurrency (line 425) | func validateConcurrency(cfg *configFile, result *ValidationResult) { function checkDeprecatedFields (line 458) | func checkDeprecatedFields(cfg *configFile, result *ValidationResult) { function getMostSpecificSchemaError (line 481) | func getMostSpecificSchemaError(errors []gojsonschema.ResultError) *Sche... function getSchemaErrorDescription (line 513) | func getSchemaErrorDescription(err gojsonschema.ResultError, allErrors [... function humanReadableSchemaType (line 528) | func humanReadableSchemaType(definition string) string { function schemaErrorSpecificity (line 549) | func schemaErrorSpecificity(err gojsonschema.ResultError) int { FILE: pkg/config/validate_test.go function TestValidateConfigFile (line 9) | func TestValidateConfigFile(t *testing.T) { function TestValidateConfigFileSuccess (line 25) | func TestValidateConfigFileSuccess(t *testing.T) { function TestValidateConfigFilePythonVersionNumerical (line 44) | func TestValidateConfigFilePythonVersionNumerical(t *testing.T) { function TestValidateConfigFileNullListsAllowed (line 63) | func TestValidateConfigFileNullListsAllowed(t *testing.T) { function TestValidateConfigFilePredictFormat (line 78) | func TestValidateConfigFilePredictFormat(t *testing.T) { function TestValidateConfigFileConcurrencyType (line 97) | func TestValidateConfigFileConcurrencyType(t *testing.T) { function TestValidateConfigFileDeprecatedPythonPackages (line 117) | func TestValidateConfigFileDeprecatedPythonPackages(t *testing.T) { function TestValidateConfigFileDeprecatedPreInstall (line 133) | func TestValidateConfigFileDeprecatedPreInstall(t *testing.T) { function TestValidateConfigFileMissingPythonVersion (line 149) | func TestValidateConfigFileMissingPythonVersion(t *testing.T) { function TestValidateConfigFileMissingPythonVersionEmptyBuild (line 161) | func TestValidateConfigFileMissingPythonVersionEmptyBuild(t *testing.T) { function TestValidateConfigFileNilBuildSkipsPythonVersionCheck (line 171) | func TestValidateConfigFileNilBuildSkipsPythonVersionCheck(t *testing.T) { function ptr (line 180) | func ptr[T any](v T) *T { return &v } FILE: pkg/config/version.go type ArgumentType (line 4) | type ArgumentType constant ArgumentTypeString (line 7) | ArgumentTypeString ArgumentType = "str" constant ArgumentTypeInt (line 8) | ArgumentTypeInt ArgumentType = "int" constant ArgumentTypeFloat (line 9) | ArgumentTypeFloat ArgumentType = "float" constant ArgumentTypeBool (line 10) | ArgumentTypeBool ArgumentType = "bool" constant ArgumentTypePath (line 11) | ArgumentTypePath ArgumentType = "Path" type RunArgument (line 15) | type RunArgument struct FILE: pkg/docker/build_secrets.go function ParseSecretsFromHost (line 14) | func ParseSecretsFromHost(workingDir string, secrets []string) (secrets.... function parseSecretFromHost (line 28) | func parseSecretFromHost(workingDir, secret string) (*secretsprovider.So... FILE: pkg/docker/buildkit.go function prepareDockerfileDir (line 25) | func prepareDockerfileDir(buildDir string, dockerfileContents string) (s... function solveOptFromImageOptions (line 34) | func solveOptFromImageOptions(buildDir string, opts command.ImageBuildOp... function newDisplay (line 155) | func newDisplay(statusCh chan *buildkitclient.SolveStatus, displayMode s... function newBuildkitAuthProvider (line 177) | func newBuildkitAuthProvider(registryHosts ...string) session.Attachable { type buildkitAuthProvider (line 187) | type buildkitAuthProvider struct method Register (line 191) | func (ap *buildkitAuthProvider) Register(server *grpc.Server) { method Credentials (line 195) | func (ap *buildkitAuthProvider) Credentials(ctx context.Context, req *... method FetchToken (line 209) | func (ap *buildkitAuthProvider) FetchToken(ctx context.Context, req *a... method GetTokenAuthority (line 213) | func (ap *buildkitAuthProvider) GetTokenAuthority(ctx context.Context,... method VerifyTokenAuthority (line 217) | func (ap *buildkitAuthProvider) VerifyTokenAuthority(ctx context.Conte... FILE: pkg/docker/command/command.go type Command (line 11) | type Command interface type ImageBuildOptions (line 35) | type ImageBuildOptions struct type RunOptions (line 54) | type RunOptions struct type Port (line 69) | type Port struct type Volume (line 74) | type Volume struct FILE: pkg/docker/command/errors.go type NotFoundError (line 9) | type NotFoundError struct method Error (line 16) | func (e *NotFoundError) Error() string { method Is (line 24) | func (e *NotFoundError) Is(target error) bool { function IsNotFoundError (line 29) | func IsNotFoundError(err error) bool { FILE: pkg/docker/command/manifest.go type Config (line 5) | type Config struct type Manifest (line 10) | type Manifest struct constant R8CogVersionEnvVarName (line 15) | R8CogVersionEnvVarName = "R8_COG_VERSION" constant R8TorchVersionEnvVarName (line 16) | R8TorchVersionEnvVarName = "R8_TORCH_VERSION" constant R8CudaVersionEnvVarName (line 17) | R8CudaVersionEnvVarName = "R8_CUDA_VERSION" constant R8CudnnVersionEnvVarName (line 18) | R8CudnnVersionEnvVarName = "R8_CUDNN_VERSION" constant R8PythonVersionEnvVarName (line 19) | R8PythonVersionEnvVarName = "R8_PYTHON_VERSION" FILE: pkg/docker/command/user_info.go type UserInfo (line 3) | type UserInfo struct FILE: pkg/docker/credential_helper_input.go type CredentialHelperInput (line 3) | type CredentialHelperInput struct FILE: pkg/docker/credentials.go function loadUserInformation (line 22) | func loadUserInformation(ctx context.Context, registryHost string) (*com... function loadAuthFromConfig (line 45) | func loadAuthFromConfig(conf *configfile.ConfigFile, registryHost string... function loadRegistryAuths (line 49) | func loadRegistryAuths(ctx context.Context, registryHosts ...string) (ma... function tryLoadAuthForHost (line 78) | func tryLoadAuthForHost(ctx context.Context, conf *configfile.ConfigFile... function loadAuthFromCredentialsStore (line 106) | func loadAuthFromCredentialsStore(ctx context.Context, credsStore string... function dockerCredentialBinary (line 149) | func dockerCredentialBinary(credsStore string) string { FILE: pkg/docker/credentials_test.go function TestLoadRegistryAuths_Fallback (line 16) | func TestLoadRegistryAuths_Fallback(t *testing.T) { function TestTryLoadAuthForHost (line 133) | func TestTryLoadAuthForHost(t *testing.T) { FILE: pkg/docker/docker.go function ptrVal (line 36) | func ptrVal[T any](v T) *T { return &v } function NewClient (line 38) | func NewClient(ctx context.Context, opts ...Option) (*apiClient, error) { type apiClient (line 88) | type apiClient struct method Pull (line 93) | func (c *apiClient) Pull(ctx context.Context, imageRef string, force b... method ContainerStop (line 132) | func (c *apiClient) ContainerStop(ctx context.Context, containerID str... method ContainerInspect (line 147) | func (c *apiClient) ContainerInspect(ctx context.Context, containerID ... method ContainerLogs (line 160) | func (c *apiClient) ContainerLogs(ctx context.Context, containerID str... method Push (line 197) | func (c *apiClient) Push(ctx context.Context, imageRef string) error { method ImageSave (line 258) | func (c *apiClient) ImageSave(ctx context.Context, imageRef string) (i... method LoadUserInformation (line 264) | func (c *apiClient) LoadUserInformation(ctx context.Context, registryH... method Inspect (line 270) | func (c *apiClient) Inspect(ctx context.Context, ref string) (*image.I... method RemoveImage (line 288) | func (c *apiClient) RemoveImage(ctx context.Context, ref string) error { method ImageExists (line 302) | func (c *apiClient) ImageExists(ctx context.Context, ref string) (bool... method ImageBuild (line 315) | func (c *apiClient) ImageBuild(ctx context.Context, options command.Im... method containerRun (line 380) | func (c *apiClient) containerRun(ctx context.Context, options command.... method Run (line 581) | func (c *apiClient) Run(ctx context.Context, options command.RunOption... method ContainerStart (line 595) | func (c *apiClient) ContainerStart(ctx context.Context, options comman... function parseGPURequest (line 604) | func parseGPURequest(opts command.RunOptions) (container.DeviceRequest, ... function shouldAttachStdin (line 639) | func shouldAttachStdin(stdin io.Reader) (attach bool, tty bool) { FILE: pkg/docker/docker_client_test.go function TestDockerClient (line 22) | func TestDockerClient(t *testing.T) { function assertImageExists (line 400) | func assertImageExists(t *testing.T, dockerClient command.Command, image... function assertNoImageExists (line 408) | func assertNoImageExists(t *testing.T, dockerClient command.Command, ima... function assertNotFoundError (line 416) | func assertNotFoundError(t *testing.T, err error, ref string, object str... FILE: pkg/docker/dockertest/command_mocks.go function NewMockCommand2 (line 19) | func NewMockCommand2(t interface { type MockCommand2 (line 32) | type MockCommand2 struct method EXPECT (line 40) | func (_m *MockCommand2) EXPECT() *MockCommand2_Expecter { method ContainerInspect (line 45) | func (_mock *MockCommand2) ContainerInspect(ctx context.Context, id st... method ContainerLogs (line 113) | func (_mock *MockCommand2) ContainerLogs(ctx context.Context, containe... method ContainerStart (line 176) | func (_mock *MockCommand2) ContainerStart(ctx context.Context, options... method ContainerStop (line 242) | func (_mock *MockCommand2) ContainerStop(ctx context.Context, containe... method ImageBuild (line 299) | func (_mock *MockCommand2) ImageBuild(ctx context.Context, options com... method ImageSave (line 365) | func (_mock *MockCommand2) ImageSave(ctx context.Context, imageRef str... method ImageExists (line 433) | func (_mock *MockCommand2) ImageExists(ctx context.Context, ref string... method Inspect (line 499) | func (_mock *MockCommand2) Inspect(ctx context.Context, ref string) (*... method LoadUserInformation (line 567) | func (_mock *MockCommand2) LoadUserInformation(ctx context.Context, re... method Pull (line 635) | func (_mock *MockCommand2) Pull(ctx context.Context, ref string, force... method Push (line 709) | func (_mock *MockCommand2) Push(ctx context.Context, ref string) error { method RemoveImage (line 766) | func (_mock *MockCommand2) RemoveImage(ctx context.Context, ref string... method Run (line 823) | func (_mock *MockCommand2) Run(ctx context.Context, options command.Ru... type MockCommand2_Expecter (line 36) | type MockCommand2_Expecter struct method ContainerInspect (line 80) | func (_e *MockCommand2_Expecter) ContainerInspect(ctx interface{}, id ... method ContainerLogs (line 138) | func (_e *MockCommand2_Expecter) ContainerLogs(ctx interface{}, contai... method ContainerStart (line 209) | func (_e *MockCommand2_Expecter) ContainerStart(ctx interface{}, optio... method ContainerStop (line 266) | func (_e *MockCommand2_Expecter) ContainerStop(ctx interface{}, contai... method ImageBuild (line 332) | func (_e *MockCommand2_Expecter) ImageBuild(ctx interface{}, options i... method ImageSave (line 400) | func (_e *MockCommand2_Expecter) ImageSave(ctx interface{}, imageRef i... method ImageExists (line 466) | func (_e *MockCommand2_Expecter) ImageExists(ctx interface{}, ref inte... method Inspect (line 534) | func (_e *MockCommand2_Expecter) Inspect(ctx interface{}, ref interfac... method LoadUserInformation (line 602) | func (_e *MockCommand2_Expecter) LoadUserInformation(ctx interface{}, ... method Pull (line 671) | func (_e *MockCommand2_Expecter) Pull(ctx interface{}, ref interface{}... method Push (line 733) | func (_e *MockCommand2_Expecter) Push(ctx interface{}, ref interface{}... method RemoveImage (line 790) | func (_e *MockCommand2_Expecter) RemoveImage(ctx interface{}, ref inte... method Run (line 847) | func (_e *MockCommand2_Expecter) Run(ctx interface{}, options interfac... type MockCommand2_ContainerInspect_Call (line 73) | type MockCommand2_ContainerInspect_Call struct method Run (line 84) | func (_c *MockCommand2_ContainerInspect_Call) Run(run func(ctx context... method Return (line 102) | func (_c *MockCommand2_ContainerInspect_Call) Return(inspectResponse *... method RunAndReturn (line 107) | func (_c *MockCommand2_ContainerInspect_Call) RunAndReturn(run func(ct... type MockCommand2_ContainerLogs_Call (line 130) | type MockCommand2_ContainerLogs_Call struct method Run (line 142) | func (_c *MockCommand2_ContainerLogs_Call) Run(run func(ctx context.Co... method Return (line 165) | func (_c *MockCommand2_ContainerLogs_Call) Return(err error) *MockComm... method RunAndReturn (line 170) | func (_c *MockCommand2_ContainerLogs_Call) RunAndReturn(run func(ctx c... type MockCommand2_ContainerStart_Call (line 202) | type MockCommand2_ContainerStart_Call struct method Run (line 213) | func (_c *MockCommand2_ContainerStart_Call) Run(run func(ctx context.C... method Return (line 231) | func (_c *MockCommand2_ContainerStart_Call) Return(s string, err error... method RunAndReturn (line 236) | func (_c *MockCommand2_ContainerStart_Call) RunAndReturn(run func(ctx ... type MockCommand2_ContainerStop_Call (line 259) | type MockCommand2_ContainerStop_Call struct method Run (line 270) | func (_c *MockCommand2_ContainerStop_Call) Run(run func(ctx context.Co... method Return (line 288) | func (_c *MockCommand2_ContainerStop_Call) Return(err error) *MockComm... method RunAndReturn (line 293) | func (_c *MockCommand2_ContainerStop_Call) RunAndReturn(run func(ctx c... type MockCommand2_ImageBuild_Call (line 325) | type MockCommand2_ImageBuild_Call struct method Run (line 336) | func (_c *MockCommand2_ImageBuild_Call) Run(run func(ctx context.Conte... method Return (line 354) | func (_c *MockCommand2_ImageBuild_Call) Return(imageID string, err err... method RunAndReturn (line 359) | func (_c *MockCommand2_ImageBuild_Call) RunAndReturn(run func(ctx cont... type MockCommand2_ImageSave_Call (line 393) | type MockCommand2_ImageSave_Call struct method Run (line 404) | func (_c *MockCommand2_ImageSave_Call) Run(run func(ctx context.Contex... method Return (line 422) | func (_c *MockCommand2_ImageSave_Call) Return(rc io.ReadCloser, err er... method RunAndReturn (line 427) | func (_c *MockCommand2_ImageSave_Call) RunAndReturn(run func(ctx conte... type MockCommand2_ImageExists_Call (line 459) | type MockCommand2_ImageExists_Call struct method Run (line 470) | func (_c *MockCommand2_ImageExists_Call) Run(run func(ctx context.Cont... method Return (line 488) | func (_c *MockCommand2_ImageExists_Call) Return(b bool, err error) *Mo... method RunAndReturn (line 493) | func (_c *MockCommand2_ImageExists_Call) RunAndReturn(run func(ctx con... type MockCommand2_Inspect_Call (line 527) | type MockCommand2_Inspect_Call struct method Run (line 538) | func (_c *MockCommand2_Inspect_Call) Run(run func(ctx context.Context,... method Return (line 556) | func (_c *MockCommand2_Inspect_Call) Return(inspectResponse *image.Ins... method RunAndReturn (line 561) | func (_c *MockCommand2_Inspect_Call) RunAndReturn(run func(ctx context... type MockCommand2_LoadUserInformation_Call (line 595) | type MockCommand2_LoadUserInformation_Call struct method Run (line 606) | func (_c *MockCommand2_LoadUserInformation_Call) Run(run func(ctx cont... method Return (line 624) | func (_c *MockCommand2_LoadUserInformation_Call) Return(userInfo *comm... method RunAndReturn (line 629) | func (_c *MockCommand2_LoadUserInformation_Call) RunAndReturn(run func... type MockCommand2_Pull_Call (line 663) | type MockCommand2_Pull_Call struct method Run (line 675) | func (_c *MockCommand2_Pull_Call) Run(run func(ctx context.Context, re... method Return (line 698) | func (_c *MockCommand2_Pull_Call) Return(inspectResponse *image.Inspec... method RunAndReturn (line 703) | func (_c *MockCommand2_Pull_Call) RunAndReturn(run func(ctx context.Co... type MockCommand2_Push_Call (line 726) | type MockCommand2_Push_Call struct method Run (line 737) | func (_c *MockCommand2_Push_Call) Run(run func(ctx context.Context, re... method Return (line 755) | func (_c *MockCommand2_Push_Call) Return(err error) *MockCommand2_Push... method RunAndReturn (line 760) | func (_c *MockCommand2_Push_Call) RunAndReturn(run func(ctx context.Co... type MockCommand2_RemoveImage_Call (line 783) | type MockCommand2_RemoveImage_Call struct method Run (line 794) | func (_c *MockCommand2_RemoveImage_Call) Run(run func(ctx context.Cont... method Return (line 812) | func (_c *MockCommand2_RemoveImage_Call) Return(err error) *MockComman... method RunAndReturn (line 817) | func (_c *MockCommand2_RemoveImage_Call) RunAndReturn(run func(ctx con... type MockCommand2_Run_Call (line 840) | type MockCommand2_Run_Call struct method Run (line 851) | func (_c *MockCommand2_Run_Call) Run(run func(ctx context.Context, opt... method Return (line 869) | func (_c *MockCommand2_Run_Call) Return(err error) *MockCommand2_Run_C... method RunAndReturn (line 874) | func (_c *MockCommand2_Run_Call) RunAndReturn(run func(ctx context.Con... FILE: pkg/docker/dockertest/helper_client.go function NewHelperClient (line 25) | func NewHelperClient(t testing.TB) *HelperClient { type HelperClient (line 67) | type HelperClient struct method Close (line 74) | func (c *HelperClient) Close() error { method PullImage (line 78) | func (c *HelperClient) PullImage(t testing.TB, ref string) error { method MustPullImage (line 103) | func (c *HelperClient) MustPullImage(t testing.TB, ref string) { method PushImage (line 108) | func (c *HelperClient) PushImage(t testing.TB, ref string) error { method MustPushImage (line 134) | func (c *HelperClient) MustPushImage(t testing.TB, ref string) { method RunContainer (line 139) | func (c *HelperClient) RunContainer(t testing.TB, imageName string) st... method StopContainer (line 180) | func (c *HelperClient) StopContainer(t testing.TB, containerID string) { method InspectImage (line 190) | func (c *HelperClient) InspectImage(t testing.TB, imageRef string) *im... method ImageExists (line 199) | func (c *HelperClient) ImageExists(t testing.TB, imageRef string) bool { method DeleteImage (line 206) | func (c *HelperClient) DeleteImage(t testing.TB, imageRef string) error { method MustDeleteImage (line 216) | func (c *HelperClient) MustDeleteImage(t testing.TB, imageRef string) { method CleanupImage (line 226) | func (c *HelperClient) CleanupImage(t testing.TB, imageRef string) { method CleanupImages (line 240) | func (c *HelperClient) CleanupImages(t testing.TB) { method InspectContainer (line 269) | func (c *HelperClient) InspectContainer(t testing.TB, containerID stri... method ImageFixture (line 278) | func (c *HelperClient) ImageFixture(t testing.TB, name string, tag str... method loadImageFixture (line 292) | func (c *HelperClient) loadImageFixture(t testing.TB, name string) *im... type imageFixture (line 341) | type imageFixture struct FILE: pkg/docker/dockertest/image.go function ImageRef (line 13) | func ImageRef(t *testing.T, label string) string { function ImageRefWithRegistry (line 21) | func ImageRefWithRegistry(t *testing.T, registryAddr string, label strin... FILE: pkg/docker/dockertest/mock_command.go type MockCommand (line 21) | type MockCommand struct method Pull (line 27) | func (c *MockCommand) Pull(ctx context.Context, image string, force bo... method Push (line 31) | func (c *MockCommand) Push(ctx context.Context, image string) error { method LoadUserInformation (line 35) | func (c *MockCommand) LoadUserInformation(ctx context.Context, registr... method CreateTarFile (line 43) | func (c *MockCommand) CreateTarFile(ctx context.Context, image string,... method CreateAptTarFile (line 53) | func (c *MockCommand) CreateAptTarFile(ctx context.Context, tmpDir str... method Inspect (line 63) | func (c *MockCommand) Inspect(ctx context.Context, ref string) (*image... method ImageExists (line 85) | func (c *MockCommand) ImageExists(ctx context.Context, ref string) (bo... method ContainerLogs (line 89) | func (c *MockCommand) ContainerLogs(ctx context.Context, containerID s... method ContainerInspect (line 93) | func (c *MockCommand) ContainerInspect(ctx context.Context, id string)... method ContainerStop (line 97) | func (c *MockCommand) ContainerStop(ctx context.Context, containerID s... method RemoveImage (line 101) | func (c *MockCommand) RemoveImage(ctx context.Context, ref string) err... method ImageBuild (line 105) | func (c *MockCommand) ImageBuild(ctx context.Context, options command.... method Run (line 109) | func (c *MockCommand) Run(ctx context.Context, options command.RunOpti... method ContainerStart (line 113) | func (c *MockCommand) ContainerStart(ctx context.Context, options comm... method ImageSave (line 117) | func (c *MockCommand) ImageSave(ctx context.Context, imageRef string) ... function NewMockCommand (line 23) | func NewMockCommand() *MockCommand { FILE: pkg/docker/dockertest/ref.go type Ref (line 11) | type Ref struct method WithTag (line 35) | func (r Ref) WithTag(tagName string) Ref { method WithDigest (line 40) | func (r Ref) WithDigest(digest string) Ref { method WithRegistry (line 45) | func (r Ref) WithRegistry(registry string) Ref { method WithoutRegistry (line 64) | func (r Ref) WithoutRegistry() Ref { method String (line 80) | func (r Ref) String() string { function NewRef (line 16) | func NewRef(t *testing.T) Ref { FILE: pkg/docker/dockertest/ref_test.go function TestRef (line 9) | func TestRef(t *testing.T) { FILE: pkg/docker/env.go constant DockerCommandEnvVarName (line 5) | DockerCommandEnvVarName = "R8_DOCKER_COMMAND" function DockerCommandFromEnvironment (line 7) | func DockerCommandFromEnvironment() string { FILE: pkg/docker/errors.go function isTagNotFoundError (line 12) | func isTagNotFoundError(err error) bool { function isAuthorizationFailedError (line 18) | func isAuthorizationFailedError(err error) bool { function isRepositoryNotFoundError (line 39) | func isRepositoryNotFoundError(err error) bool { function isMissingDeviceDriverError (line 45) | func isMissingDeviceDriverError(err error) bool { function isNetworkError (line 52) | func isNetworkError(err error) bool { FILE: pkg/docker/host.go function determineDockerHost (line 16) | func determineDockerHost() (string, error) { function dockerHostFromContext (line 44) | func dockerHostFromContext(contextName string) (string, string, error) { FILE: pkg/docker/host_unix.go constant defaultDockerHost (line 6) | defaultDockerHost = "unix:///var/run/docker.sock" FILE: pkg/docker/host_windows.go constant defaultDockerHost (line 4) | defaultDockerHost = "npipe:////.pipe/docker_engine" FILE: pkg/docker/login.go function SaveLoginToken (line 18) | func SaveLoginToken(ctx context.Context, registryHost string, username s... function saveAuthToConfig (line 27) | func saveAuthToConfig(conf *configfile.ConfigFile, registryHost string, ... function saveAuthToCredentialsStore (line 39) | func saveAuthToCredentialsStore(ctx context.Context, credsStore string, ... FILE: pkg/docker/options.go type clientOptions (line 5) | type clientOptions struct type Option (line 10) | type Option function WithAuthConfig (line 12) | func WithAuthConfig(authConfig registry.AuthConfig) Option { function WithHost (line 18) | func WithHost(host string) Option { FILE: pkg/docker/progress.go type ProgressWriter (line 20) | type ProgressWriter struct method Write (line 48) | func (p *ProgressWriter) Write(id, status string, current, total int64) { method WriteStatus (line 62) | func (p *ProgressWriter) WriteStatus(id, status string) { method writeMessage (line 70) | func (p *ProgressWriter) writeMessage(msg jsonmessage.JSONMessage) { method Close (line 87) | func (p *ProgressWriter) Close() { function NewProgressWriter (line 29) | func NewProgressWriter() *ProgressWriter { FILE: pkg/docker/push.go type BuildInfo (line 13) | type BuildInfo struct function Push (line 18) | func Push(ctx context.Context, image string, projectDir string, command ... FILE: pkg/docker/run.go function Run (line 19) | func Run(ctx context.Context, dockerClient command.Command, options comm... function RunWithIO (line 23) | func RunWithIO(ctx context.Context, dockerClient command.Command, option... function RunDaemon (line 32) | func RunDaemon(ctx context.Context, dockerClient command.Command, option... function GetHostPortForContainer (line 37) | func GetHostPortForContainer(ctx context.Context, dockerCommand command.... FILE: pkg/docker/run_test.go function TestGetHostPortForContainer (line 14) | func TestGetHostPortForContainer(t *testing.T) { FILE: pkg/docker/standard_push.go function StandardPush (line 9) | func StandardPush(ctx context.Context, image string, command command.Com... FILE: pkg/docker/standard_push_test.go function TestStandardPush (line 11) | func TestStandardPush(t *testing.T) { FILE: pkg/dockercontext/build_tempdir.go function CogBuildArtifactsDirPath (line 11) | func CogBuildArtifactsDirPath(dir string) (string, error) { function CogTempDir (line 20) | func CogTempDir(dir string, contextDir string) (string, error) { function BuildCogTempDir (line 28) | func BuildCogTempDir(dir string, subDir string) (string, error) { function BuildTempDir (line 39) | func BuildTempDir(dir string) (string, error) { FILE: pkg/dockercontext/build_tempdir_test.go function TestBuildCogTempDir (line 10) | func TestBuildCogTempDir(t *testing.T) { FILE: pkg/dockercontext/directories.go constant StandardBuildDirectory (line 5) | StandardBuildDirectory = "." constant ContextBuildDir (line 7) | ContextBuildDir = "context" constant AptBuildContextName (line 8) | AptBuildContextName = "apt" constant RequirementsBuildContextName (line 9) | RequirementsBuildContextName = "requirements" constant SrcBuildContextName (line 10) | SrcBuildContextName = "src" FILE: pkg/dockerfile/base.go constant MinimumCUDAVersion (line 16) | MinimumCUDAVersion = "11.6" constant MinimumPythonVersion (line 17) | MinimumPythonVersion = "3.10" constant MinimumTorchVersion (line 18) | MinimumTorchVersion = "1.13.1" constant CogBaseImageName (line 19) | CogBaseImageName = "cog-base" type CUDAVersion (line 53) | type CUDAVersion struct type PyTorchVersion (line 57) | type PyTorchVersion struct type PythonVersion (line 61) | type PythonVersion struct type AvailableBaseImageConfigurations (line 67) | type AvailableBaseImageConfigurations struct type BaseImageConfiguration (line 71) | type BaseImageConfiguration struct method MarshalJSON (line 85) | func (b BaseImageConfiguration) MarshalJSON() ([]byte, error) { type BaseImageGenerator (line 77) | type BaseImageGenerator struct method GenerateDockerfile (line 184) | func (g *BaseImageGenerator) GenerateDockerfile(ctx context.Context) (... method makeConfig (line 205) | func (g *BaseImageGenerator) makeConfig() (*config.Config, error) { method pythonPackages (line 222) | func (g *BaseImageGenerator) pythonPackages() []string { method runStatements (line 260) | func (g *BaseImageGenerator) runStatements() []config.RunItem { function BaseImageConfigurations (line 110) | func BaseImageConfigurations() []BaseImageConfiguration { function NewBaseImageGenerator (line 167) | func NewBaseImageGenerator(ctx context.Context, client registry.Client, ... function baseImageComponentNormalisation (line 264) | func baseImageComponentNormalisation(cudaVersion string, pythonVersion s... function BaseImageName (line 290) | func BaseImageName(cudaVersion string, pythonVersion string, torchVersio... function BaseImageConfigurationExists (line 312) | func BaseImageConfigurationExists(ctx context.Context, client registry.C... function isVersionCompatible (line 343) | func isVersionCompatible(confVersion, requestedVersion string) bool { FILE: pkg/dockerfile/base_test.go function TestBaseImageName (line 14) | func TestBaseImageName(t *testing.T) { function TestGenerateDockerfile (line 37) | func TestGenerateDockerfile(t *testing.T) { function TestBaseImageNameWithVersionModifier (line 59) | func TestBaseImageNameWithVersionModifier(t *testing.T) { function TestBaseImageConfigurationExists (line 64) | func TestBaseImageConfigurationExists(t *testing.T) { function TestBaseImageConfigurationExistsNoTorch (line 76) | func TestBaseImageConfigurationExistsNoTorch(t *testing.T) { function TestBaseImageConfigurationExistsNoCUDA (line 87) | func TestBaseImageConfigurationExistsNoCUDA(t *testing.T) { function TestIsVersionCompatible (line 99) | func TestIsVersionCompatible(t *testing.T) { function TestPythonPackages (line 104) | func TestPythonPackages(t *testing.T) { function TestInvalidBaseImage (line 122) | func TestInvalidBaseImage(t *testing.T) { function TestBaseImageConfigurationNoTorchPythonVersionDoesNotExist (line 129) | func TestBaseImageConfigurationNoTorchPythonVersionDoesNotExist(t *testi... FILE: pkg/dockerfile/cacert.go constant CACertEnvVar (line 15) | CACertEnvVar = "COG_CA_CERT" constant CACertFilename (line 18) | CACertFilename = "cog-ca-cert.crt" constant CACertContainerPath (line 21) | CACertContainerPath = "/usr/local/share/ca-certificates/" + CACertFilename constant SystemCertBundle (line 24) | SystemCertBundle = "/etc/ssl/certs/ca-certificates.crt" function ReadCACert (line 38) | func ReadCACert() ([]byte, error) { function readCACertFile (line 69) | func readCACertFile(path string) ([]byte, error) { function readCACertDirectory (line 78) | func readCACertDirectory(dir string) ([]byte, error) { function validatePEM (line 120) | func validatePEM(data []byte) ([]byte, error) { function GenerateCACertInstall (line 144) | func GenerateCACertInstall(certData []byte, writeTemp func(filename stri... FILE: pkg/dockerfile/cacert_test.go constant testCertPEM (line 13) | testCertPEM = `-----BEGIN CERTIFICATE----- constant testCertPEM2 (line 23) | testCertPEM2 = `-----BEGIN CERTIFICATE----- function TestReadCACert_NotSet (line 33) | func TestReadCACert_NotSet(t *testing.T) { function TestReadCACert_FilePath (line 41) | func TestReadCACert_FilePath(t *testing.T) { function TestReadCACert_Directory (line 55) | func TestReadCACert_Directory(t *testing.T) { function TestReadCACert_InlinePEM (line 74) | func TestReadCACert_InlinePEM(t *testing.T) { function TestReadCACert_Base64EncodedPEM (line 83) | func TestReadCACert_Base64EncodedPEM(t *testing.T) { function TestReadCACert_InvalidPEM (line 93) | func TestReadCACert_InvalidPEM(t *testing.T) { function TestReadCACert_MissingFile (line 102) | func TestReadCACert_MissingFile(t *testing.T) { function TestReadCACert_EmptyDirectory (line 111) | func TestReadCACert_EmptyDirectory(t *testing.T) { function TestReadCACert_InvalidFileInDirectory (line 121) | func TestReadCACert_InvalidFileInDirectory(t *testing.T) { function TestReadCACert_TrimsWhitespace (line 134) | func TestReadCACert_TrimsWhitespace(t *testing.T) { function TestGenerateCACertInstall (line 144) | func TestGenerateCACertInstall(t *testing.T) { function TestGenerateCACertInstall_EmptyData (line 168) | func TestGenerateCACertInstall_EmptyData(t *testing.T) { FILE: pkg/dockerfile/env.go function envLineFromConfig (line 10) | func envLineFromConfig(c *config.Config) (string, error) { FILE: pkg/dockerfile/generator.go type Generator (line 9) | type Generator interface FILE: pkg/dockerfile/generator_factory.go function NewGenerator (line 9) | func NewGenerator(config *config.Config, dir string, configFilename stri... FILE: pkg/dockerfile/generator_factory_test.go function TestGeneratorFactoryStandardGenerator (line 13) | func TestGeneratorFactoryStandardGenerator(t *testing.T) { FILE: pkg/dockerfile/standard_generator.go constant DockerignoreHeader (line 23) | DockerignoreHeader = `# generated by replicate/cog constant LDConfigCacheBuildCommand (line 45) | LDConfigCacheBuildCommand = "RUN find / -type f -name \"*python*.so\" -p... constant StripDebugSymbolsCommand (line 46) | StripDebugSymbolsCommand = "find / -type f -name \"*python*.so\" -not -n... constant CFlags (line 47) | CFlags = "ENV CFLAGS=\"-O3 -funroll-loops -fno-strict-aliasing -flto -S\"" constant UVVersion (line 48) | UVVersion = "0.9.26" constant uvCacheMount (line 49) | uvCacheMount = "--mount=type=cache,target=/root/.cache/uv" constant uvPip (line 50) | uvPip = "uv pip" constant PrecompilePythonCommand (line 51) | PrecompilePythonCommand = "RUN find / -type f -name \"*.py[co]\" -delete... constant STANDARD_GENERATOR_NAME (line 52) | STANDARD_GENERATOR_NAME = "STANDARD_GENERATOR" type StandardGenerator (line 54) | type StandardGenerator struct method SetUseCudaBaseImage (line 131) | func (g *StandardGenerator) SetUseCudaBaseImage(argumentValue string) { method SetUseCogBaseImage (line 136) | func (g *StandardGenerator) SetUseCogBaseImage(useCogBaseImage bool) { method SetUseCogBaseImagePtr (line 141) | func (g *StandardGenerator) SetUseCogBaseImagePtr(useCogBaseImage *boo... method IsUsingCogBaseImage (line 145) | func (g *StandardGenerator) IsUsingCogBaseImage() bool { method SetStrip (line 153) | func (g *StandardGenerator) SetStrip(strip bool) { method SetPrecompile (line 157) | func (g *StandardGenerator) SetPrecompile(precompile bool) { method GenerateInitialSteps (line 161) | func (g *StandardGenerator) GenerateInitialSteps(ctx context.Context) ... method GenerateModelBase (line 243) | func (g *StandardGenerator) GenerateModelBase(ctx context.Context) (st... method GenerateDockerfileWithoutSeparateWeights (line 259) | func (g *StandardGenerator) GenerateDockerfileWithoutSeparateWeights(c... method GenerateModelBaseWithSeparateWeights (line 280) | func (g *StandardGenerator) GenerateModelBaseWithSeparateWeights(ctx c... method cogEnvVars (line 326) | func (g *StandardGenerator) cogEnvVars() []string { method cpCogYaml (line 340) | func (g *StandardGenerator) cpCogYaml() string { method generateForWeights (line 348) | func (g *StandardGenerator) generateForWeights() (string, []string, []... method Cleanup (line 376) | func (g *StandardGenerator) Cleanup() error { method BaseImage (line 383) | func (g *StandardGenerator) BaseImage(ctx context.Context) (string, er... method Name (line 402) | func (g *StandardGenerator) Name() string { method BuildDir (line 406) | func (g *StandardGenerator) BuildDir() (string, error) { method BuildContexts (line 410) | func (g *StandardGenerator) BuildContexts() (map[string]string, error) { method preamble (line 414) | func (g *StandardGenerator) preamble() string { method installTini (line 421) | func (g *StandardGenerator) installTini() string { method aptInstalls (line 441) | func (g *StandardGenerator) aptInstalls() (string, error) { method installPython (line 459) | func (g *StandardGenerator) installPython() (string, error) { method installUV (line 466) | func (g *StandardGenerator) installUV() string { method installPythonCUDA (line 471) | func (g *StandardGenerator) installPythonCUDA() (string, error) { method resolveCogWheelConfigs (line 504) | func (g *StandardGenerator) resolveCogWheelConfigs() error { method isLegacySDKVersion (line 556) | func (g *StandardGenerator) isLegacySDKVersion() bool { method installCog (line 572) | func (g *StandardGenerator) installCog() (string, error) { method installCogFromPyPI (line 657) | func (g *StandardGenerator) installCogFromPyPI(config *wheels.WheelCon... method installWheelFromURL (line 672) | func (g *StandardGenerator) installWheelFromURL(url string) (string, e... method installWheelFromFile (line 702) | func (g *StandardGenerator) installWheelFromFile(path string) (string,... method installCogletWheel (line 741) | func (g *StandardGenerator) installCogletWheel(config *wheels.WheelCon... method installCogletFromPyPI (line 756) | func (g *StandardGenerator) installCogletFromPyPI(config *wheels.Wheel... method installCogletFromURL (line 771) | func (g *StandardGenerator) installCogletFromURL(url string) (string, ... method installCogletFromFile (line 781) | func (g *StandardGenerator) installCogletFromFile(path string) (string... method filterManagedPackages (line 806) | func (g *StandardGenerator) filterManagedPackages(reqContents string) ... method pipInstalls (line 859) | func (g *StandardGenerator) pipInstalls() (string, error) { method runCommands (line 912) | func (g *StandardGenerator) runCommands() (string, error) { method envVars (line 945) | func (g *StandardGenerator) envVars() (string, error) { method installCACert (line 951) | func (g *StandardGenerator) installCACert() (string, error) { method writeTemp (line 964) | func (g *StandardGenerator) writeTemp(filename string, contents []byte... method GenerateWeightsManifest (line 994) | func (g *StandardGenerator) GenerateWeightsManifest(ctx context.Contex... method determineBaseImageName (line 1023) | func (g *StandardGenerator) determineBaseImageName(ctx context.Context... function NewStandardGenerator (line 94) | func NewStandardGenerator(config *config.Config, dir string, configFilen... function makeDockerignoreForWeights (line 365) | func makeDockerignoreForWeights(dirs, files []string) string { constant cogletMinSDKVersion (line 551) | cogletMinSDKVersion = "0.17.0" function joinStringsWithoutLineSpace (line 975) | func joinStringsWithoutLineSpace(chunks []string) string { function filterEmpty (line 984) | func filterEmpty(list []string) []string { function stripPatchVersion (line 1049) | func stripPatchVersion(versionString string) (string, bool, error) { FILE: pkg/dockerfile/standard_generator_test.go function testTini (line 19) | func testTini() string { function testInstallCog (line 34) | func testInstallCog(stripped bool) string { function pypiWheels (line 48) | func pypiWheels(gen *StandardGenerator) { function testInstallPython (line 53) | func testInstallPython(version string) string { function TestGenerateEmptyCPU (line 70) | func TestGenerateEmptyCPU(t *testing.T) { function TestGenerateEmptyGPU (line 108) | func TestGenerateEmptyGPU(t *testing.T) { function TestGenerateFullCPU (line 146) | func TestGenerateFullCPU(t *testing.T) { function TestGenerateFullGPU (line 205) | func TestGenerateFullGPU(t *testing.T) { function TestPreInstall (line 264) | func TestPreInstall(t *testing.T) { function TestPythonRequirements (line 306) | func TestPythonRequirements(t *testing.T) { type mockFileInfo (line 330) | type mockFileInfo struct method Size (line 334) | func (mfi mockFileInfo) Size() int64 { method Name (line 337) | func (mfi mockFileInfo) Name() string { method Mode (line 340) | func (mfi mockFileInfo) Mode() os.FileMode { method ModTime (line 343) | func (mfi mockFileInfo) ModTime() time.Time { method IsDir (line 346) | func (mfi mockFileInfo) IsDir() bool { method Sys (line 349) | func (mfi mockFileInfo) Sys() any { constant sizeThreshold (line 353) | sizeThreshold = 10 * 1024 * 1024 function TestGenerateWithLargeModels (line 355) | func TestGenerateWithLargeModels(t *testing.T) { function TestGenerateDockerfileWithoutSeparateWeights (line 463) | func TestGenerateDockerfileWithoutSeparateWeights(t *testing.T) { function TestGenerateEmptyCPUWithCogBaseImage (line 500) | func TestGenerateEmptyCPUWithCogBaseImage(t *testing.T) { function TestGeneratePythonCPUWithCogBaseImage (line 534) | func TestGeneratePythonCPUWithCogBaseImage(t *testing.T) { function TestGenerateFullGPUWithCogBaseImage (line 585) | func TestGenerateFullGPUWithCogBaseImage(t *testing.T) { function TestGenerateTorchWithStrippedModifiedVersion (line 650) | func TestGenerateTorchWithStrippedModifiedVersion(t *testing.T) { function TestGenerateWithStrip (line 707) | func TestGenerateWithStrip(t *testing.T) { function TestGenerateDoesNotContainDangerousCFlags (line 765) | func TestGenerateDoesNotContainDangerousCFlags(t *testing.T) { function TestGenerateWithPrecompile (line 800) | func TestGenerateWithPrecompile(t *testing.T) { function TestGenerateWithCoglet (line 860) | func TestGenerateWithCoglet(t *testing.T) { function TestCOGWheelDefault (line 923) | func TestCOGWheelDefault(t *testing.T) { function TestCOGWheelEnvPyPI (line 951) | func TestCOGWheelEnvPyPI(t *testing.T) { function TestCOGWheelEnvPyPIWithVersion (line 979) | func TestCOGWheelEnvPyPIWithVersion(t *testing.T) { function TestCOGWheelEnvURL (line 1007) | func TestCOGWheelEnvURL(t *testing.T) { function TestCOGWheelEnvFile (line 1036) | func TestCOGWheelEnvFile(t *testing.T) { function TestCogletStrippedFromRequirements (line 1069) | func TestCogletStrippedFromRequirements(t *testing.T) { function TestInstallCogWithSDKVersion (line 1094) | func TestInstallCogWithSDKVersion(t *testing.T) { function TestInstallCogWithPreReleaseSDKVersion (line 1120) | func TestInstallCogWithPreReleaseSDKVersion(t *testing.T) { function TestInstallCogSDKVersionBelowMinimum (line 1146) | func TestInstallCogSDKVersionBelowMinimum(t *testing.T) { function TestCOGSDKWheelEnvVarOverridesSDKVersion (line 1170) | func TestCOGSDKWheelEnvVarOverridesSDKVersion(t *testing.T) { FILE: pkg/dockerfile/version_check.go function parse (line 11) | func parse(s string) (string, string, string) { function CheckMajorOnly (line 23) | func CheckMajorOnly(s string) bool { function CheckMajorMinorOnly (line 28) | func CheckMajorMinorOnly(s string) bool { function CheckMajorMinorPatch (line 33) | func CheckMajorMinorPatch(s string) bool { FILE: pkg/dockerignore/dockerignore.go constant DockerIgnoreFilename (line 13) | DockerIgnoreFilename = ".dockerignore" function CreateMatcher (line 15) | func CreateMatcher(dir string) (*ignore.GitIgnore, error) { function Walk (line 32) | func Walk(root string, ignoreMatcher *ignore.GitIgnore, fn filepath.Walk... function readDockerIgnore (line 58) | func readDockerIgnore(dockerIgnorePath string) ([]string, error) { FILE: pkg/dockerignore/dockerignore_test.go function TestWalk (line 11) | func TestWalk(t *testing.T) { FILE: pkg/env/env.go constant SchemeEnvVarName (line 5) | SchemeEnvVarName = "R8_SCHEME" constant WebHostEnvVarName (line 6) | WebHostEnvVarName = "R8_WEB_HOST" constant APIHostEnvVarName (line 7) | APIHostEnvVarName = "R8_API_HOST" constant PytorchHostEnvVarName (line 8) | PytorchHostEnvVarName = "R8_PYTORCH_HOST" function SchemeFromEnvironment (line 10) | func SchemeFromEnvironment() string { function WebHostFromEnvironment (line 18) | func WebHostFromEnvironment() string { function APIHostFromEnvironment (line 26) | func APIHostFromEnvironment() string { function PytorchHostFromEnvironment (line 34) | func PytorchHostFromEnvironment() string { FILE: pkg/env/env_test.go function TestSchemeFromEnvironment (line 9) | func TestSchemeFromEnvironment(t *testing.T) { function TestWebHostFromEnvironment (line 15) | func TestWebHostFromEnvironment(t *testing.T) { FILE: pkg/errors/errors.go constant CodeConfigNotFound (line 4) | CodeConfigNotFound = "CONFIG_NOT_FOUND" type CodedError (line 9) | type CodedError interface type codedError (line 13) | type codedError struct method Error (line 18) | func (e *codedError) Error() string { method Code (line 22) | func (e *codedError) Code() string { function ConfigNotFound (line 29) | func ConfigNotFound(msg string) error { function IsConfigNotFound (line 38) | func IsConfigNotFound(err error) bool { function Code (line 43) | func Code(err error) string { FILE: pkg/global/global.go constant DefaultReplicateRegistryHost (line 6) | DefaultReplicateRegistryHost = "r8.im" constant ReplicateWebsiteHost (line 7) | ReplicateWebsiteHost = "replicate.com" function getDefaultRegistryHost (line 23) | func getDefaultRegistryHost() string { FILE: pkg/http/client.go constant UserAgentHeader (line 12) | UserAgentHeader = "User-Agent" constant BearerHeaderPrefix (line 13) | BearerHeaderPrefix = "Bearer " function ProvideHTTPClient (line 15) | func ProvideHTTPClient(ctx context.Context, dockerCommand command.Comman... FILE: pkg/http/client_test.go function TestClientDecoratesUserAgent (line 13) | func TestClientDecoratesUserAgent(t *testing.T) { FILE: pkg/http/transport.go constant AuthorizationHeader (line 8) | AuthorizationHeader = "Authorization" type Transport (line 10) | type Transport struct method RoundTrip (line 16) | func (t *Transport) RoundTrip(req *http.Request) (*http.Response, erro... FILE: pkg/http/transport_test.go function TestTransportAddsHeaders (line 12) | func TestTransportAddsHeaders(t *testing.T) { function TestTransportOnlyAddsHeaderIfMissing (line 33) | func TestTransportOnlyAddsHeaderIfMissing(t *testing.T) { function TestTransportSendsErrorWithMissingToken (line 56) | func TestTransportSendsErrorWithMissingToken(t *testing.T) { FILE: pkg/http/user_agent.go function UserAgent (line 10) | func UserAgent() string { FILE: pkg/http/user_agent_test.go function TestUserAgent (line 10) | func TestUserAgent(t *testing.T) { FILE: pkg/image/build.go constant dockerignoreBackupPath (line 37) | dockerignoreBackupPath = ".dockerignore.cog.bak" constant weightsManifestPath (line 38) | weightsManifestPath = ".cog/cache/weights_manifest.json" constant bundledSchemaFile (line 39) | bundledSchemaFile = ".cog/openapi_schema.json" function Build (line 46) | func Build( function BuildAddLabelsAndSchemaToImage (line 415) | func BuildAddLabelsAndSchemaToImage(ctx context.Context, dockerClient co... constant staticSchemaGenMinSDKVersion (line 440) | staticSchemaGenMinSDKVersion = "0.17.0" function canUseStaticSchemaGen (line 448) | func canUseStaticSchemaGen(cfg *config.Config) bool { function resolveSDKVersion (line 477) | func resolveSDKVersion(cfg *config.Config) string { function generateStaticSchema (line 496) | func generateStaticSchema(cfg *config.Config, dir string) ([]byte, error) { function writeAndValidateSchema (line 506) | func writeAndValidateSchema(schemaJSON []byte) error { function isGitWorkTree (line 526) | func isGitWorkTree(ctx context.Context, dir string) bool { function gitHead (line 538) | func gitHead(ctx context.Context, dir string) (string, error) { function gitTag (line 558) | func gitTag(ctx context.Context, dir string) (string, error) { function buildWeightsImage (line 578) | func buildWeightsImage(ctx context.Context, dockerClient command.Command... function buildRunnerImage (line 599) | func buildRunnerImage(ctx context.Context, dockerClient command.Command,... function makeDockerignoreForWeightsImage (line 623) | func makeDockerignoreForWeightsImage() error { function writeDockerignore (line 634) | func writeDockerignore(contents string) error { function backupDockerignore (line 647) | func backupDockerignore() error { function restoreDockerignore (line 660) | func restoreDockerignore() error { function checkCompatibleDockerIgnore (line 676) | func checkCompatibleDockerIgnore(dir string) error { FILE: pkg/image/build_test.go function gitRun (line 21) | func gitRun(ctx context.Context, argv []string, t *testing.T) { function setupGitWorkTree (line 31) | func setupGitWorkTree(t *testing.T) string { function TestIsGitWorkTree (line 52) | func TestIsGitWorkTree(t *testing.T) { function TestGitHead (line 60) | func TestGitHead(t *testing.T) { function TestGitTag (line 92) | func TestGitTag(t *testing.T) { function TestCanUseStaticSchemaGen (line 123) | func TestCanUseStaticSchemaGen(t *testing.T) { FILE: pkg/image/config.go function CogConfigFromManifest (line 14) | func CogConfigFromManifest(ctx context.Context, manifest *image.InspectR... function friendlyName (line 32) | func friendlyName(manifest *image.InspectResponse) string { FILE: pkg/image/openapi_schema.go function GenerateOpenAPISchema (line 20) | func GenerateOpenAPISchema(ctx context.Context, dockerClient command.Com... FILE: pkg/image/pip_freeze.go function GeneratePipFreeze (line 14) | func GeneratePipFreeze(ctx context.Context, dockerClient command.Command... FILE: pkg/model/artifact.go type ArtifactType (line 6) | type ArtifactType method String (line 16) | func (t ArtifactType) String() string { constant ArtifactTypeImage (line 10) | ArtifactTypeImage ArtifactType = iota + 1 constant ArtifactTypeWeight (line 12) | ArtifactTypeWeight type ArtifactSpec (line 30) | type ArtifactSpec interface type Artifact (line 37) | type Artifact interface FILE: pkg/model/artifact_image.go type ImageSource (line 14) | type ImageSource constant ImageSourceLocal (line 17) | ImageSourceLocal ImageSource = "local" constant ImageSourceRemote (line 18) | ImageSourceRemote ImageSource = "remote" constant ImageSourceBuild (line 19) | ImageSourceBuild ImageSource = "build" type Platform (line 23) | type Platform struct type ImageSpecOption (line 42) | type ImageSpecOption function WithImageSecrets (line 45) | func WithImageSecrets(secrets []string) ImageSpecOption { function WithImageNoCache (line 52) | func WithImageNoCache(noCache bool) ImageSpecOption { type ImageSpec (line 64) | type ImageSpec struct method Type (line 85) | func (s *ImageSpec) Type() ArtifactType { return ArtifactTypeImage } method Name (line 88) | func (s *ImageSpec) Name() string { return s.name } function NewImageSpec (line 73) | func NewImageSpec(name, imageName string, opts ...ImageSpecOption) *Imag... type ImageArtifact (line 98) | type ImageArtifact struct method Type (line 121) | func (a *ImageArtifact) Type() ArtifactType { return ArtifactTypeImage } method Name (line 124) | func (a *ImageArtifact) Name() string { return a.name } method Descriptor (line 127) | func (a *ImageArtifact) Descriptor() v1.Descriptor { return a.descript... method IsCogModel (line 134) | func (a *ImageArtifact) IsCogModel() bool { method CogVersion (line 143) | func (a *ImageArtifact) CogVersion() string { method Config (line 151) | func (a *ImageArtifact) Config() string { method OpenAPISchema (line 159) | func (a *ImageArtifact) OpenAPISchema() string { method ParsedConfig (line 169) | func (a *ImageArtifact) ParsedConfig() (*config.Config, error) { method ParsedOpenAPISchema (line 185) | func (a *ImageArtifact) ParsedOpenAPISchema() (*openapi3.T, error) { method ToModel (line 201) | func (a *ImageArtifact) ToModel() (*Model, error) { function NewImageArtifact (line 112) | func NewImageArtifact(name string, desc v1.Descriptor, reference string)... FILE: pkg/model/artifact_image_test.go function TestImageSpec_ImplementsArtifactSpec (line 10) | func TestImageSpec_ImplementsArtifactSpec(t *testing.T) { function TestImageSpec_Fields (line 19) | func TestImageSpec_Fields(t *testing.T) { function TestImageSpec_DefaultFields (line 30) | func TestImageSpec_DefaultFields(t *testing.T) { function TestImageArtifact_ImplementsArtifact (line 38) | func TestImageArtifact_ImplementsArtifact(t *testing.T) { function TestImageArtifact_Reference (line 52) | func TestImageArtifact_Reference(t *testing.T) { FILE: pkg/model/artifact_test.go function TestArtifactType_String (line 9) | func TestArtifactType_String(t *testing.T) { function TestArtifactType_Values (line 27) | func TestArtifactType_Values(t *testing.T) { FILE: pkg/model/artifact_weight.go constant MediaTypeWeightArtifact (line 12) | MediaTypeWeightArtifact = "application/vnd.cog.weight.v1" constant MediaTypeWeightConfig (line 14) | MediaTypeWeightConfig = "application/vnd.cog.weight.config.v1+json" constant MediaTypeWeightLayer (line 16) | MediaTypeWeightLayer = "application/vnd.cog.weight.layer.v1" constant MediaTypeWeightLayerGzip (line 18) | MediaTypeWeightLayerGzip = "application/vnd.cog.weight.layer.v1+gzip" constant MediaTypeWeightLayerZstd (line 20) | MediaTypeWeightLayerZstd = "application/vnd.cog.weight.layer.v1+zstd" constant AnnotationWeightName (line 25) | AnnotationWeightName = "vnd.cog.weight.name" constant AnnotationWeightDest (line 26) | AnnotationWeightDest = "vnd.cog.weight.dest" constant AnnotationWeightDigestOriginal (line 27) | AnnotationWeightDigestOriginal = "vnd.cog.weight.digest.original" constant AnnotationWeightSizeUncompressed (line 28) | AnnotationWeightSizeUncompressed = "vnd.cog.weight.size.uncompressed" type WeightSpec (line 33) | type WeightSpec struct method Type (line 51) | func (s *WeightSpec) Type() ArtifactType { return ArtifactTypeWeight } method Name (line 54) | func (s *WeightSpec) Name() string { return s.name } function NewWeightSpec (line 42) | func NewWeightSpec(name, source, target string) *WeightSpec { type WeightArtifact (line 58) | type WeightArtifact struct method Type (line 82) | func (a *WeightArtifact) Type() ArtifactType { return ArtifactTypeWeig... method Name (line 85) | func (a *WeightArtifact) Name() string { return a.name } method Descriptor (line 88) | func (a *WeightArtifact) Descriptor() v1.Descriptor { return a.descrip... function NewWeightArtifact (line 71) | func NewWeightArtifact(name string, desc v1.Descriptor, filePath, target... type WeightConfig (line 93) | type WeightConfig struct FILE: pkg/model/artifact_weight_test.go function TestWeightSpec_ImplementsArtifactSpec (line 12) | func TestWeightSpec_ImplementsArtifactSpec(t *testing.T) { function TestWeightSpec_Fields (line 21) | func TestWeightSpec_Fields(t *testing.T) { function TestWeightArtifact_ImplementsArtifact (line 28) | func TestWeightArtifact_ImplementsArtifact(t *testing.T) { function TestWeightArtifact_Fields (line 49) | func TestWeightArtifact_Fields(t *testing.T) { function TestWeightConfig_JSONRoundTrip (line 68) | func TestWeightConfig_JSONRoundTrip(t *testing.T) { function TestWeightMediaTypeConstants (line 100) | func TestWeightMediaTypeConstants(t *testing.T) { FILE: pkg/model/builder.go type Builder (line 7) | type Builder interface FILE: pkg/model/builder_test.go type mockBuilder (line 12) | type mockBuilder struct method Build (line 16) | func (m *mockBuilder) Build(ctx context.Context, spec ArtifactSpec) (A... function TestBuilderInterface_Satisfiable (line 20) | func TestBuilderInterface_Satisfiable(t *testing.T) { FILE: pkg/model/errors_test.go function TestSentinelErrors (line 11) | func TestSentinelErrors(t *testing.T) { FILE: pkg/model/factory.go type Factory (line 13) | type Factory interface type DockerfileFactory (line 24) | type DockerfileFactory struct method Name (line 35) | func (f *DockerfileFactory) Name() string { method Build (line 40) | func (f *DockerfileFactory) Build(ctx context.Context, src *Source, op... function NewDockerfileFactory (line 30) | func NewDockerfileFactory(docker command.Command, registry registry.Clie... function defaultFactory (line 80) | func defaultFactory(docker command.Command, registry registry.Client) Fa... FILE: pkg/model/factory_test.go function TestDockerfileFactory_Name (line 12) | func TestDockerfileFactory_Name(t *testing.T) { function TestDockerfileFactory_ImplementsInterface (line 21) | func TestDockerfileFactory_ImplementsInterface(t *testing.T) { function TestDefaultFactory_ReturnsDockerfileFactory (line 29) | func TestDefaultFactory_ReturnsDockerfileFactory(t *testing.T) { FILE: pkg/model/format.go function OCIIndexEnabled (line 9) | func OCIIndexEnabled() bool { FILE: pkg/model/format_test.go function TestOCIIndexEnabled_Default (line 9) | func TestOCIIndexEnabled_Default(t *testing.T) { function TestOCIIndexEnabled_Enabled (line 14) | func TestOCIIndexEnabled_Enabled(t *testing.T) { function TestOCIIndexEnabled_OtherValue (line 19) | func TestOCIIndexEnabled_OtherValue(t *testing.T) { FILE: pkg/model/hash.go function hashFile (line 11) | func hashFile(path string) (digest string, size int64, err error) { FILE: pkg/model/image_builder.go type ImageBuilder (line 16) | type ImageBuilder struct method Build (line 37) | func (b *ImageBuilder) Build(ctx context.Context, spec ArtifactSpec) (... function NewImageBuilder (line 24) | func NewImageBuilder(factory Factory, docker command.Command, source *So... FILE: pkg/model/image_builder_test.go function TestImageBuilder_HappyPath (line 16) | func TestImageBuilder_HappyPath(t *testing.T) { function TestImageBuilder_ErrorWrongSpecType (line 74) | func TestImageBuilder_ErrorWrongSpecType(t *testing.T) { function TestImageBuilder_ErrorFactoryBuildFails (line 85) | func TestImageBuilder_ErrorFactoryBuildFails(t *testing.T) { function TestImageBuilder_ErrorInspectFails (line 102) | func TestImageBuilder_ErrorInspectFails(t *testing.T) { function TestImageBuilder_ImplementsBuilderInterface (line 128) | func TestImageBuilder_ImplementsBuilderInterface(t *testing.T) { FILE: pkg/model/image_pusher.go type ImagePusher (line 30) | type ImagePusher struct method Push (line 73) | func (p *ImagePusher) Push(ctx context.Context, artifact *ImageArtifac... method canOCIPush (line 106) | func (p *ImagePusher) canOCIPush() bool { method ociPush (line 112) | func (p *ImagePusher) ociPush(ctx context.Context, imageRef string, op... method pushImage (line 158) | func (p *ImagePusher) pushImage(ctx context.Context, imageRef string, ... method pushLayers (line 179) | func (p *ImagePusher) pushLayers(ctx context.Context, repo string, img... method pushLayer (line 205) | func (p *ImagePusher) pushLayer(ctx context.Context, repo string, laye... method pushConfig (line 243) | func (p *ImagePusher) pushConfig(ctx context.Context, repo string, img... function newImagePusher (line 36) | func newImagePusher(docker command.Command, reg registry.Client) *ImageP... type imagePushOptions (line 44) | type imagePushOptions struct type ImagePushOption (line 50) | type ImagePushOption function WithProgressFn (line 53) | func WithProgressFn(fn func(PushProgress)) ImagePushOption { function WithOnFallback (line 62) | func WithOnFallback(fn func()) ImagePushOption { function shouldFallbackToDocker (line 270) | func shouldFallbackToDocker(err error) bool { function sanitizeError (line 292) | func sanitizeError(err error) error { type configBlobLayer (line 302) | type configBlobLayer struct method Digest (line 307) | func (c *configBlobLayer) Digest() (v1.Hash, error) { method DiffID (line 314) | func (c *configBlobLayer) DiffID() (v1.Hash, error) { method Compressed (line 318) | func (c *configBlobLayer) Compressed() (io.ReadCloser, error) { method Uncompressed (line 322) | func (c *configBlobLayer) Uncompressed() (io.ReadCloser, error) { method Size (line 326) | func (c *configBlobLayer) Size() (int64, error) { method MediaType (line 330) | func (c *configBlobLayer) MediaType() (types.MediaType, error) { FILE: pkg/model/image_pusher_test.go type ociMockClient (line 27) | type ociMockClient struct method WriteLayer (line 36) | func (m *ociMockClient) WriteLayer(_ context.Context, opts registry.Wr... method PushImage (line 57) | func (m *ociMockClient) PushImage(_ context.Context, ref string, _ v1.... method Inspect (line 67) | func (m *ociMockClient) Inspect(context.Context, string, *registry.Pla... method GetImage (line 70) | func (m *ociMockClient) GetImage(context.Context, string, *registry.Pl... method Exists (line 73) | func (m *ociMockClient) Exists(context.Context, string) (bool, error) ... method GetDescriptor (line 74) | func (m *ociMockClient) GetDescriptor(context.Context, string) (v1.Des... method PushIndex (line 77) | func (m *ociMockClient) PushIndex(context.Context, string, v1.ImageInd... function testArtifact (line 80) | func testArtifact(ref string) *ImageArtifact { function fakeImageSaveFunc (line 86) | func fakeImageSaveFunc(img v1.Image, tagStr string) func(context.Context... function TestImagePusher_Push (line 105) | func TestImagePusher_Push(t *testing.T) { function TestImagePusher_PushArtifact (line 242) | func TestImagePusher_PushArtifact(t *testing.T) { function TestImagePusher_Fallback (line 301) | func TestImagePusher_Fallback(t *testing.T) { function TestShouldFallbackToDocker (line 434) | func TestShouldFallbackToDocker(t *testing.T) { function TestSanitizeError (line 463) | func TestSanitizeError(t *testing.T) { function TestImagePusher_OnFallback (line 509) | func TestImagePusher_OnFallback(t *testing.T) { function TestConfigBlobLayer (line 564) | func TestConfigBlobLayer(t *testing.T) { function TestGetPushConcurrency (line 617) | func TestGetPushConcurrency(t *testing.T) { FILE: pkg/model/image_test.go function TestImage_IsCogModel (line 12) | func TestImage_IsCogModel(t *testing.T) { function TestImage_CogVersion (line 65) | func TestImage_CogVersion(t *testing.T) { function TestImage_Config (line 109) | func TestImage_Config(t *testing.T) { function TestImage_OpenAPISchema (line 139) | func TestImage_OpenAPISchema(t *testing.T) { function TestImageSource_Values (line 169) | func TestImageSource_Values(t *testing.T) { function TestLabelKeys (line 176) | func TestLabelKeys(t *testing.T) { function TestImage_ParsedConfig (line 188) | func TestImage_ParsedConfig(t *testing.T) { function TestImage_ToModel (line 267) | func TestImage_ToModel(t *testing.T) { function TestImage_ParsedOpenAPISchema (line 357) | func TestImage_ParsedOpenAPISchema(t *testing.T) { FILE: pkg/model/index.go type Index (line 5) | type Index struct type IndexManifest (line 17) | type IndexManifest struct type ManifestType (line 33) | type ManifestType constant ManifestTypeImage (line 37) | ManifestTypeImage ManifestType = "image" constant ManifestTypeWeights (line 39) | ManifestTypeWeights ManifestType = "weights" constant AnnotationReferenceType (line 44) | AnnotationReferenceType = "vnd.cog.reference.type" constant AnnotationReferenceDigest (line 45) | AnnotationReferenceDigest = "vnd.cog.reference.digest" constant AnnotationValueWeights (line 51) | AnnotationValueWeights = "weights" constant PlatformUnknown (line 57) | PlatformUnknown = "unknown" FILE: pkg/model/index_factory.go type IndexBuilder (line 13) | type IndexBuilder struct method SetImageDescriptor (line 33) | func (b *IndexBuilder) SetImageDescriptor(desc v1.Descriptor, platform... method AddWeightDescriptor (line 41) | func (b *IndexBuilder) AddWeightDescriptor(desc v1.Descriptor, imageDi... method BuildFromDescriptors (line 53) | func (b *IndexBuilder) BuildFromDescriptors() (v1.ImageIndex, error) { type weightDescEntry (line 20) | type weightDescEntry struct function NewIndexBuilder (line 28) | func NewIndexBuilder() *IndexBuilder { type descriptorAppendable (line 106) | type descriptorAppendable struct method MediaType (line 110) | func (d *descriptorAppendable) MediaType() (types.MediaType, error) { method Digest (line 114) | func (d *descriptorAppendable) Digest() (v1.Hash, error) { method Size (line 118) | func (d *descriptorAppendable) Size() (int64, error) { FILE: pkg/model/index_factory_test.go function TestIndexBuilder_BuildFromDescriptors (line 11) | func TestIndexBuilder_BuildFromDescriptors(t *testing.T) { FILE: pkg/model/index_test.go function TestModel_IsBundle (line 11) | func TestModel_IsBundle(t *testing.T) { function TestManifestType (line 37) | func TestManifestType(t *testing.T) { FILE: pkg/model/model.go type Model (line 12) | type Model struct method HasGPU (line 32) | func (m *Model) HasGPU() bool { method SchemaJSON (line 37) | func (m *Model) SchemaJSON() ([]byte, error) { method ImageRef (line 45) | func (m *Model) ImageRef() string { method IsBundle (line 53) | func (m *Model) IsBundle() bool { method GetImageArtifact (line 59) | func (m *Model) GetImageArtifact() *ImageArtifact { method WeightArtifacts (line 69) | func (m *Model) WeightArtifacts() []*WeightArtifact { method ArtifactsByType (line 80) | func (m *Model) ArtifactsByType(t ArtifactType) []Artifact { FILE: pkg/model/model_test.go function TestModel_HasGPU (line 14) | func TestModel_HasGPU(t *testing.T) { function TestModel_SchemaJSON (line 50) | func TestModel_SchemaJSON(t *testing.T) { function TestModel_ImageRef (line 88) | func TestModel_ImageRef(t *testing.T) { function TestModel_GetImageArtifact (line 123) | func TestModel_GetImageArtifact(t *testing.T) { function TestModel_WeightArtifacts (line 175) | func TestModel_WeightArtifacts(t *testing.T) { function TestModel_ArtifactsByType (line 210) | func TestModel_ArtifactsByType(t *testing.T) { FILE: pkg/model/options.go type BuildOptions (line 7) | type BuildOptions struct method WithDefaults (line 73) | func (o BuildOptions) WithDefaults(src *Source) BuildOptions { FILE: pkg/model/options_test.go function TestBuildOptions_WithDefaults_ImageName (line 11) | func TestBuildOptions_WithDefaults_ImageName(t *testing.T) { function TestBuildOptions_WithDefaults_PreservesExplicitImageName (line 24) | func TestBuildOptions_WithDefaults_PreservesExplicitImageName(t *testing... function TestBuildOptions_WithDefaults_ProgressOutput (line 36) | func TestBuildOptions_WithDefaults_ProgressOutput(t *testing.T) { function TestBuildOptions_WithDefaults_PreservesExplicitProgressOutput (line 48) | func TestBuildOptions_WithDefaults_PreservesExplicitProgressOutput(t *te... function TestBuildOptions_WithDefaults_NilBuildConfig (line 60) | func TestBuildOptions_WithDefaults_NilBuildConfig(t *testing.T) { function TestBuildOptions_WithDefaults_NilConfig (line 74) | func TestBuildOptions_WithDefaults_NilConfig(t *testing.T) { function TestBuildOptions_AllFieldsPreserved (line 88) | func TestBuildOptions_AllFieldsPreserved(t *testing.T) { function TestBuildOptions_WeightsLockPath (line 129) | func TestBuildOptions_WeightsLockPath(t *testing.T) { FILE: pkg/model/push_helpers.go constant DefaultPushConcurrency (line 17) | DefaultPushConcurrency = 5 constant envPushConcurrency (line 20) | envPushConcurrency = "COG_PUSH_CONCURRENCY" function GetPushConcurrency (line 25) | func GetPushConcurrency() int { type PushProgress (line 36) | type PushProgress struct function writeLayerWithProgress (line 51) | func writeLayerWithProgress(ctx context.Context, reg registry.Client, op... FILE: pkg/model/pusher.go type PushOptions (line 17) | type PushOptions struct type BundlePusher (line 50) | type BundlePusher struct method Push (line 70) | func (p *BundlePusher) Push(ctx context.Context, m *Model, opts PushOp... method pushWeights (line 142) | func (p *BundlePusher) pushWeights(ctx context.Context, repo string, w... function NewBundlePusher (line 60) | func NewBundlePusher(docker command.Command, reg registry.Client) *Bundl... function repoFromReference (line 170) | func repoFromReference(ref string) string { FILE: pkg/model/pusher_test.go function TestBundlePusher_Push (line 22) | func TestBundlePusher_Push(t *testing.T) { function TestResolver_Push (line 420) | func TestResolver_Push(t *testing.T) { function TestPushOptions (line 537) | func TestPushOptions(t *testing.T) { function TestRepoFromReference (line 558) | func TestRepoFromReference(t *testing.T) { FILE: pkg/model/ref.go type ParseOption (line 12) | type ParseOption type parseOptions (line 14) | type parseOptions struct function Insecure (line 21) | func Insecure() ParseOption { function WithDefaultRegistry (line 29) | func WithDefaultRegistry(registry string) ParseOption { function WithDefaultTag (line 37) | func WithDefaultTag(tag string) ParseOption { type ParsedRef (line 44) | type ParsedRef struct method String (line 71) | func (p *ParsedRef) String() string { method Registry (line 76) | func (p *ParsedRef) Registry() string { method Repository (line 81) | func (p *ParsedRef) Repository() string { method Tag (line 86) | func (p *ParsedRef) Tag() string { method Digest (line 94) | func (p *ParsedRef) Digest() string { method IsTag (line 102) | func (p *ParsedRef) IsTag() bool { method IsDigest (line 108) | func (p *ParsedRef) IsDigest() bool { method IsReplicate (line 114) | func (p *ParsedRef) IsReplicate() bool { function ParseRef (line 53) | func ParseRef(ref string, opts ...ParseOption) (*ParsedRef, error) { FILE: pkg/model/ref_test.go function TestParseRef (line 9) | func TestParseRef(t *testing.T) { function TestParsedRef_IsTag (line 146) | func TestParsedRef_IsTag(t *testing.T) { function TestParsedRef_String (line 158) | func TestParsedRef_String(t *testing.T) { function TestParseOptions (line 190) | func TestParseOptions(t *testing.T) { FILE: pkg/model/ref_types.go type Ref (line 9) | type Ref interface method Resolve (line 17) | func (r *Resolver) Resolve(ctx context.Context, ref Ref) (*Model, error) { type TagRef (line 28) | type TagRef struct method resolve (line 42) | func (t *TagRef) resolve(ctx context.Context, r *Resolver) (*Model, er... function FromTag (line 34) | func FromTag(ref string) (*TagRef, error) { type LocalRef (line 53) | type LocalRef struct method resolve (line 67) | func (l *LocalRef) resolve(ctx context.Context, r *Resolver) (*Model, ... function FromLocal (line 59) | func FromLocal(ref string) (*LocalRef, error) { type RemoteRef (line 77) | type RemoteRef struct method resolve (line 91) | func (rr *RemoteRef) resolve(ctx context.Context, r *Resolver) (*Model... function FromRemote (line 83) | func FromRemote(ref string) (*RemoteRef, error) { type BuildRef (line 101) | type BuildRef struct method resolve (line 113) | func (b *BuildRef) resolve(ctx context.Context, r *Resolver) (*Model, ... function FromBuild (line 109) | func FromBuild(src *Source, opts BuildOptions) *BuildRef { FILE: pkg/model/ref_types_test.go function TestFromTag_ValidRef (line 21) | func TestFromTag_ValidRef(t *testing.T) { function TestFromTag_ValidRefWithRegistry (line 30) | func TestFromTag_ValidRefWithRegistry(t *testing.T) { function TestFromTag_InvalidRef (line 39) | func TestFromTag_InvalidRef(t *testing.T) { function TestFromLocal_ValidRef (line 51) | func TestFromLocal_ValidRef(t *testing.T) { function TestFromLocal_InvalidRef (line 60) | func TestFromLocal_InvalidRef(t *testing.T) { function TestFromRemote_ValidRef (line 72) | func TestFromRemote_ValidRef(t *testing.T) { function TestFromRemote_InvalidRef (line 81) | func TestFromRemote_InvalidRef(t *testing.T) { function TestFromBuild (line 93) | func TestFromBuild(t *testing.T) { function TestFromBuild_NilSource (line 111) | func TestFromBuild_NilSource(t *testing.T) { function TestTagRef_Resolve_Success (line 123) | func TestTagRef_Resolve_Success(t *testing.T) { function TestTagRef_Resolve_FallsBackToLocal (line 152) | func TestTagRef_Resolve_FallsBackToLocal(t *testing.T) { function TestLocalRef_Resolve_Success (line 196) | func TestLocalRef_Resolve_Success(t *testing.T) { function TestLocalRef_Resolve_NotFound (line 226) | func TestLocalRef_Resolve_NotFound(t *testing.T) { function TestRemoteRef_Resolve_Success (line 253) | func TestRemoteRef_Resolve_Success(t *testing.T) { function TestRemoteRef_Resolve_NotFound (line 284) | func TestRemoteRef_Resolve_NotFound(t *testing.T) { function TestBuildRef_Resolve_Success (line 306) | func TestBuildRef_Resolve_Success(t *testing.T) { function TestBuildRef_Resolve_BuildError (line 350) | func TestBuildRef_Resolve_BuildError(t *testing.T) { function TestResolver_Resolve_DispatchesCorrectly (line 376) | func TestResolver_Resolve_DispatchesCorrectly(t *testing.T) { FILE: pkg/model/resolver.go type Option (line 18) | type Option type options (line 20) | type options struct function defaultOptions (line 27) | func defaultOptions() *options { function LocalOnly (line 33) | func LocalOnly() Option { function RemoteOnly (line 43) | func RemoteOnly() Option { function PreferLocal (line 52) | func PreferLocal() Option { function PreferRemote (line 62) | func PreferRemote() Option { function WithPlatform (line 71) | func WithPlatform(p *registry.Platform) Option { type Resolver (line 78) | type Resolver struct method WithFactory (line 96) | func (r *Resolver) WithFactory(f Factory) *Resolver { method Inspect (line 105) | func (r *Resolver) Inspect(ctx context.Context, ref *ParsedRef, opts .... method InspectByID (line 144) | func (r *Resolver) InspectByID(ctx context.Context, id string) (*Model... method Pull (line 169) | func (r *Resolver) Pull(ctx context.Context, ref *ParsedRef, opts ...O... method Build (line 213) | func (r *Resolver) Build(ctx context.Context, src *Source, opts BuildO... method Push (line 272) | func (r *Resolver) Push(ctx context.Context, m *Model, opts PushOption... method loadLocal (line 294) | func (r *Resolver) loadLocal(ctx context.Context, ref *ParsedRef) (*Mo... method loadRemote (line 306) | func (r *Resolver) loadRemote(ctx context.Context, ref *ParsedRef, pla... method modelFromImage (line 319) | func (r *Resolver) modelFromImage(img *ImageArtifact, cfg *config.Conf... method modelFromInspect (line 335) | func (r *Resolver) modelFromInspect(ref *ParsedRef, resp *image.Inspec... method modelFromManifest (line 352) | func (r *Resolver) modelFromManifest(ref *ParsedRef, manifest *registr... method modelFromIndex (line 375) | func (r *Resolver) modelFromIndex(ref *ParsedRef, manifest *registry.M... function NewResolver (line 86) | func NewResolver(docker command.Command, reg registry.Client) *Resolver { function isOCIIndex (line 435) | func isOCIIndex(mr *registry.ManifestResult) bool { function findWeightsManifest (line 441) | func findWeightsManifest(manifests []registry.PlatformManifest) *registr... function findImageManifest (line 454) | func findImageManifest(manifests []registry.PlatformManifest, platform *... function isNotFoundError (line 474) | func isNotFoundError(err error) bool { FILE: pkg/model/resolver_test.go type mockDocker (line 24) | type mockDocker struct method Inspect (line 31) | func (m *mockDocker) Inspect(ctx context.Context, ref string) (*image.... method Pull (line 38) | func (m *mockDocker) Pull(ctx context.Context, ref string, force bool)... method Push (line 45) | func (m *mockDocker) Push(ctx context.Context, ref string) error { method LoadUserInformation (line 52) | func (m *mockDocker) LoadUserInformation(ctx context.Context, registry... method ImageExists (line 56) | func (m *mockDocker) ImageExists(ctx context.Context, ref string) (boo... method ContainerLogs (line 60) | func (m *mockDocker) ContainerLogs(ctx context.Context, containerID st... method ContainerInspect (line 64) | func (m *mockDocker) ContainerInspect(ctx context.Context, id string) ... method ContainerStop (line 68) | func (m *mockDocker) ContainerStop(ctx context.Context, containerID st... method RemoveImage (line 72) | func (m *mockDocker) RemoveImage(ctx context.Context, ref string) error { method ImageBuild (line 76) | func (m *mockDocker) ImageBuild(ctx context.Context, options command.I... method Run (line 80) | func (m *mockDocker) Run(ctx context.Context, options command.RunOptio... method ContainerStart (line 84) | func (m *mockDocker) ContainerStart(ctx context.Context, options comma... method ImageSave (line 88) | func (m *mockDocker) ImageSave(ctx context.Context, imageRef string) (... type mockRegistry (line 96) | type mockRegistry struct method Inspect (line 105) | func (m *mockRegistry) Inspect(ctx context.Context, ref string, platfo... method GetImage (line 112) | func (m *mockRegistry) GetImage(ctx context.Context, ref string, platf... method GetDescriptor (line 119) | func (m *mockRegistry) GetDescriptor(ctx context.Context, ref string) ... method Exists (line 126) | func (m *mockRegistry) Exists(ctx context.Context, ref string) (bool, ... method PushImage (line 130) | func (m *mockRegistry) PushImage(ctx context.Context, ref string, img ... method PushIndex (line 137) | func (m *mockRegistry) PushIndex(ctx context.Context, ref string, idx ... method WriteLayer (line 144) | func (m *mockRegistry) WriteLayer(ctx context.Context, opts registry.W... type mockFactory (line 153) | type mockFactory struct method Build (line 158) | func (f *mockFactory) Build(ctx context.Context, src *Source, opts Bui... method Name (line 165) | func (f *mockFactory) Name() string { function TestNewResolver (line 169) | func TestNewResolver(t *testing.T) { function TestResolver_WithFactory (line 179) | func TestResolver_WithFactory(t *testing.T) { function TestResolver_Inspect_LocalOnly_Found (line 194) | func TestResolver_Inspect_LocalOnly_Found(t *testing.T) { function TestResolver_Inspect_LocalOnly_NotFound (line 227) | func TestResolver_Inspect_LocalOnly_NotFound(t *testing.T) { function TestResolver_Inspect_RemoteOnly_Found (line 250) | func TestResolver_Inspect_RemoteOnly_Found(t *testing.T) { function TestResolver_Inspect_RemoteOnly_NotFound (line 283) | func TestResolver_Inspect_RemoteOnly_NotFound(t *testing.T) { function TestResolver_Inspect_RemoteOnly_NotCogModel (line 301) | func TestResolver_Inspect_RemoteOnly_NotCogModel(t *testing.T) { function TestResolver_Inspect_PreferLocal_FoundLocally (line 327) | func TestResolver_Inspect_PreferLocal_FoundLocally(t *testing.T) { function TestResolver_Inspect_PreferLocal_Fallback (line 367) | func TestResolver_Inspect_PreferLocal_Fallback(t *testing.T) { function TestResolver_Inspect_PreferLocal_NoFallbackOnRealError (line 402) | func TestResolver_Inspect_PreferLocal_NoFallbackOnRealError(t *testing.T) { function TestResolver_Inspect_PreferRemote_FoundRemotely (line 426) | func TestResolver_Inspect_PreferRemote_FoundRemotely(t *testing.T) { function TestResolver_Inspect_PreferRemote_Fallback (line 461) | func TestResolver_Inspect_PreferRemote_Fallback(t *testing.T) { function TestResolver_Inspect_PreferRemote_NoFallbackOnRealError (line 501) | func TestResolver_Inspect_PreferRemote_NoFallbackOnRealError(t *testing.... function TestResolver_Inspect_WithPlatform (line 525) | func TestResolver_Inspect_WithPlatform(t *testing.T) { function TestResolver_Inspect_ParsesConfigFromLabels (line 555) | func TestResolver_Inspect_ParsesConfigFromLabels(t *testing.T) { function TestResolver_Inspect_InvalidConfigJSON (line 587) | func TestResolver_Inspect_InvalidConfigJSON(t *testing.T) { function TestResolver_Inspect_NoConfigLabel_ReturnsErrNotCogModel (line 615) | func TestResolver_Inspect_NoConfigLabel_ReturnsErrNotCogModel(t *testing... function TestResolver_Inspect_NotCogModel (line 644) | func TestResolver_Inspect_NotCogModel(t *testing.T) { function TestResolver_InspectByID_Found (line 673) | func TestResolver_InspectByID_Found(t *testing.T) { function TestResolver_InspectByID_FullSHA (line 707) | func TestResolver_InspectByID_FullSHA(t *testing.T) { function TestResolver_InspectByID_NotCogModel (line 736) | func TestResolver_InspectByID_NotCogModel(t *testing.T) { function TestResolver_InspectByID_NotFound (line 762) | func TestResolver_InspectByID_NotFound(t *testing.T) { function TestResolver_Pull_AlreadyLocal (line 781) | func TestResolver_Pull_AlreadyLocal(t *testing.T) { function TestResolver_Pull_NotLocal_PullsAndReturns (line 817) | func TestResolver_Pull_NotLocal_PullsAndReturns(t *testing.T) { function TestResolver_Pull_NotCogModel (line 871) | func TestResolver_Pull_NotCogModel(t *testing.T) { function TestResolver_Pull_LocalOnly_NotFound (line 919) | func TestResolver_Pull_LocalOnly_NotFound(t *testing.T) { function TestResolver_Pull_PullFails (line 940) | func TestResolver_Pull_PullFails(t *testing.T) { function TestResolver_Pull_LocalInspectRealError (line 960) | func TestResolver_Pull_LocalInspectRealError(t *testing.T) { function TestResolver_Build_NoWeightsManifestWithoutWeights (line 985) | func TestResolver_Build_NoWeightsManifestWithoutWeights(t *testing.T) { function TestResolver_Build_PopulatesArtifacts (line 1022) | func TestResolver_Build_PopulatesArtifacts(t *testing.T) { function TestResolver_Build_PopulatesWeightArtifacts (line 1075) | func TestResolver_Build_PopulatesWeightArtifacts(t *testing.T) { function TestResolver_Build_WithWeightsLoadsManifest (line 1153) | func TestResolver_Build_WithWeightsLoadsManifest(t *testing.T) { function TestIndexDetectionHelpers (line 1215) | func TestIndexDetectionHelpers(t *testing.T) { function TestIsNotFoundError (line 1301) | func TestIsNotFoundError(t *testing.T) { FILE: pkg/model/source.go type Source (line 13) | type Source struct method ArtifactSpecs (line 77) | func (s *Source) ArtifactSpecs() []ArtifactSpec { function NewSource (line 23) | func NewSource(configPath string) (*Source, error) { function NewSourceFromConfig (line 66) | func NewSourceFromConfig(cfg *config.Config, projectDir string) *Source { FILE: pkg/model/source_test.go function TestNewSourceFromConfig (line 11) | func TestNewSourceFromConfig(t *testing.T) { function TestNewSourceFromConfig_NilConfig (line 27) | func TestNewSourceFromConfig_NilConfig(t *testing.T) { function TestSource_ArtifactSpecs_NoWeights (line 35) | func TestSource_ArtifactSpecs_NoWeights(t *testing.T) { function TestSource_ArtifactSpecs_WithWeights (line 57) | func TestSource_ArtifactSpecs_WithWeights(t *testing.T) { function TestSource_ArtifactSpecs_EmptyImageName (line 92) | func TestSource_ArtifactSpecs_EmptyImageName(t *testing.T) { function TestSource_ArtifactSpecs_NilConfig (line 106) | func TestSource_ArtifactSpecs_NilConfig(t *testing.T) { FILE: pkg/model/weight_builder.go type WeightBuilder (line 15) | type WeightBuilder struct method Build (line 34) | func (b *WeightBuilder) Build(ctx context.Context, spec ArtifactSpec) ... method findCachedEntry (line 107) | func (b *WeightBuilder) findCachedEntry(name string, fileSize int64) *... method updateLockfile (line 125) | func (b *WeightBuilder) updateLockfile(ws *WeightSpec, digest string, ... function NewWeightBuilder (line 23) | func NewWeightBuilder(source *Source, cogVersion, lockPath string) *Weig... FILE: pkg/model/weight_builder_test.go function TestWeightBuilder_HappyPath (line 16) | func TestWeightBuilder_HappyPath(t *testing.T) { function TestWeightBuilder_WritesLockfile (line 71) | func TestWeightBuilder_WritesLockfile(t *testing.T) { function TestWeightBuilder_UpdatesExistingLockfile (line 111) | func TestWeightBuilder_UpdatesExistingLockfile(t *testing.T) { function TestWeightBuilder_CacheHit (line 157) | func TestWeightBuilder_CacheHit(t *testing.T) { function TestWeightBuilder_CacheMiss_SizeChanged (line 199) | func TestWeightBuilder_CacheMiss_SizeChanged(t *testing.T) { function TestWeightBuilder_ErrorWrongSpecType (line 237) | func TestWeightBuilder_ErrorWrongSpecType(t *testing.T) { function TestWeightBuilder_ErrorFileNotFound (line 250) | func TestWeightBuilder_ErrorFileNotFound(t *testing.T) { function TestWeightBuilder_ErrorContextCancelled (line 262) | func TestWeightBuilder_ErrorContextCancelled(t *testing.T) { function TestWeightBuilder_ImplementsBuilderInterface (line 280) | func TestWeightBuilder_ImplementsBuilderInterface(t *testing.T) { FILE: pkg/model/weight_pusher.go type WeightPushOptions (line 24) | type WeightPushOptions struct type WeightRetryEvent (line 33) | type WeightRetryEvent struct type WeightPushResult (line 47) | type WeightPushResult struct type WeightPusher (line 58) | type WeightPusher struct method Push (line 71) | func (p *WeightPusher) Push(ctx context.Context, repo string, artifact... function NewWeightPusher (line 63) | func NewWeightPusher(reg registry.Client) *WeightPusher { function buildWeightImage (line 168) | func buildWeightImage(artifact *WeightArtifact) (v1.Image, error) { function descriptorFromImage (line 204) | func descriptorFromImage(img v1.Image) (v1.Descriptor, error) { type weightOCIManifest (line 225) | type weightOCIManifest struct type weightManifestImage (line 241) | type weightManifestImage struct method RawConfigFile (line 250) | func (w *weightManifestImage) RawConfigFile() ([]byte, error) { method Digest (line 255) | func (w *weightManifestImage) Digest() (v1.Hash, error) { method ArtifactType (line 268) | func (w *weightManifestImage) ArtifactType() (string, error) { method Manifest (line 273) | func (w *weightManifestImage) Manifest() (*v1.Manifest, error) { method RawManifest (line 297) | func (w *weightManifestImage) RawManifest() ([]byte, error) { constant weightTagPrefix (line 325) | weightTagPrefix = "weights-" function WeightTag (line 331) | func WeightTag(name, digest string) string { FILE: pkg/model/weight_pusher_test.go function TestWeightPusher_Push_ReturnsErrorForNilArtifact (line 20) | func TestWeightPusher_Push_ReturnsErrorForNilArtifact(t *testing.T) { function TestWeightPusher_Push_ReturnsErrorForMissingFile (line 30) | func TestWeightPusher_Push_ReturnsErrorForMissingFile(t *testing.T) { function TestWeightPusher_Push_PushesCorrectOCIArtifact (line 48) | func TestWeightPusher_Push_PushesCorrectOCIArtifact(t *testing.T) { function TestWeightPusher_Push_PropagatesPushError (line 125) | func TestWeightPusher_Push_PropagatesPushError(t *testing.T) { function TestWeightPusher_Push_RawManifestContainsArtifactType (line 152) | func TestWeightPusher_Push_RawManifestContainsArtifactType(t *testing.T) { function TestWeightPusher_Push_ReturnsErrorForEmptyRepo (line 202) | func TestWeightPusher_Push_ReturnsErrorForEmptyRepo(t *testing.T) { function TestWeightPusher_Push_ReportsProgressViaWriteLayer (line 224) | func TestWeightPusher_Push_ReportsProgressViaWriteLayer(t *testing.T) { function TestWeightPusher_Push_ForwardsRetryCallback (line 282) | func TestWeightPusher_Push_ForwardsRetryCallback(t *testing.T) { function TestWeightPusher_Push_WriteLayerErrorReported (line 332) | func TestWeightPusher_Push_WriteLayerErrorReported(t *testing.T) { function TestWeightPusher_Push_PropagatesContextCancellation (line 359) | func TestWeightPusher_Push_PropagatesContextCancellation(t *testing.T) { FILE: pkg/model/weights.go type WeightFile (line 5) | type WeightFile struct FILE: pkg/model/weights_lock.go constant WeightsLockFilename (line 12) | WeightsLockFilename = "weights.lock" type WeightsLock (line 16) | type WeightsLock struct method Save (line 44) | func (wl *WeightsLock) Save(path string) error { function ParseWeightsLock (line 26) | func ParseWeightsLock(data []byte) (*WeightsLock, error) { function LoadWeightsLock (line 35) | func LoadWeightsLock(path string) (*WeightsLock, error) { FILE: pkg/model/weights_lock_test.go function TestWeightsLock (line 13) | func TestWeightsLock(t *testing.T) { FILE: pkg/model/weights_test.go function TestWeightFile (line 9) | func TestWeightFile(t *testing.T) { FILE: pkg/path/path.go function TrimExt (line 9) | func TrimExt(s string) string { function IsExtInteger (line 13) | func IsExtInteger(ext string) bool { FILE: pkg/path/path_test.go function TestTrimExt (line 9) | func TestTrimExt(t *testing.T) { function TestIsExtInteger (line 14) | func TestIsExtInteger(t *testing.T) { FILE: pkg/predict/api.go type HelpResponse (line 5) | type HelpResponse struct FILE: pkg/predict/input.go type Input (line 19) | type Input struct type Inputs (line 28) | type Inputs method toMap (line 149) | func (inputs *Inputs) toMap() (map[string]any, error) { function NewInputs (line 30) | func NewInputs(keyVals map[string][]string, schema *openapi3.T) (Inputs,... function NewInputsForMode (line 34) | func NewInputsForMode(keyVals map[string][]string, schema *openapi3.T, i... function NewInputsWithBaseDir (line 136) | func NewInputsWithBaseDir(keyVals map[string]string, baseDir string) Inp... function fileToDataURL (line 192) | func fileToDataURL(filePath string) (string, error) { function resolveSchemaType (line 212) | func resolveSchemaType(s *openapi3.Schema) *openapi3.Schema { FILE: pkg/predict/predictor.go type status (line 21) | type status type HealthcheckResponse (line 23) | type HealthcheckResponse struct type RequestContext (line 27) | type RequestContext struct type Request (line 31) | type Request struct type Response (line 37) | type Response struct type ValidationErrorResponse (line 43) | type ValidationErrorResponse struct type Predictor (line 51) | type Predictor struct method Start (line 75) | func (p *Predictor) Start(ctx context.Context, logsWriter io.Writer, t... method waitForContainerReady (line 103) | func (p *Predictor) waitForContainerReady(ctx context.Context, timeout... method Stop (line 165) | func (p *Predictor) Stop(ctx context.Context) error { method Predict (line 169) | func (p *Predictor) Predict(inputs Inputs, context RequestContext) (*R... method GetSchema (line 219) | func (p *Predictor) GetSchema() (*openapi3.T, error) { method endpoint (line 235) | func (p *Predictor) endpoint() string { method url (line 242) | func (p *Predictor) url() string { method buildInputValidationErrorMessage (line 246) | func (p *Predictor) buildInputValidationErrorMessage(errorResponse *Va... function NewPredictor (line 61) | func NewPredictor(ctx context.Context, runOptions command.RunOptions, is... FILE: pkg/provider/generic/generic.go type GenericProvider (line 18) | type GenericProvider struct method Name (line 25) | func (p *GenericProvider) Name() string { method MatchesRegistry (line 29) | func (p *GenericProvider) MatchesRegistry(host string) bool { method Login (line 33) | func (p *GenericProvider) Login(ctx context.Context, opts provider.Log... method PostPush (line 73) | func (p *GenericProvider) PostPush(ctx context.Context, opts provider.... function New (line 21) | func New() *GenericProvider { FILE: pkg/provider/generic/generic_test.go function TestGenericProvider_Name (line 13) | func TestGenericProvider_Name(t *testing.T) { function TestGenericProvider_MatchesRegistry (line 18) | func TestGenericProvider_MatchesRegistry(t *testing.T) { function TestGenericProvider_Login (line 27) | func TestGenericProvider_Login(t *testing.T) { function TestGenericProvider_PostPush (line 41) | func TestGenericProvider_PostPush(t *testing.T) { FILE: pkg/provider/provider.go type PushOptions (line 10) | type PushOptions struct type LoginOptions (line 16) | type LoginOptions struct type Provider (line 22) | type Provider interface FILE: pkg/provider/registry.go function DefaultRegistry (line 13) | func DefaultRegistry() *Registry { function RegisterProvider (line 24) | func RegisterProvider(p Provider) { type Registry (line 29) | type Registry struct method Register (line 44) | func (r *Registry) Register(p Provider) { method ForImage (line 52) | func (r *Registry) ForImage(image string) Provider { method ForHost (line 59) | func (r *Registry) ForHost(host string) Provider { function NewRegistry (line 35) | func NewRegistry() *Registry { function ExtractHost (line 80) | func ExtractHost(image string) string { FILE: pkg/provider/registry_test.go type mockProvider (line 11) | type mockProvider struct method Name (line 16) | func (m *mockProvider) Name() string { method MatchesRegistry (line 20) | func (m *mockProvider) MatchesRegistry(host string) bool { method Login (line 24) | func (m *mockProvider) Login(ctx context.Context, opts LoginOptions) e... method PrePush (line 28) | func (m *mockProvider) PrePush(ctx context.Context, opts PushOptions) ... method PostPush (line 32) | func (m *mockProvider) PostPush(ctx context.Context, opts PushOptions,... function TestRegistry_ForHost (line 36) | func TestRegistry_ForHost(t *testing.T) { function TestRegistry_ForImage (line 71) | func TestRegistry_ForImage(t *testing.T) { function TestRegistry_NoProviders (line 108) | func TestRegistry_NoProviders(t *testing.T) { function TestExtractHost (line 115) | func TestExtractHost(t *testing.T) { FILE: pkg/provider/replicate/replicate.go type ReplicateProvider (line 25) | type ReplicateProvider struct method Name (line 32) | func (p *ReplicateProvider) Name() string { method MatchesRegistry (line 36) | func (p *ReplicateProvider) MatchesRegistry(host string) bool { method Login (line 45) | func (p *ReplicateProvider) Login(ctx context.Context, opts provider.L... method PostPush (line 82) | func (p *ReplicateProvider) PostPush(ctx context.Context, opts provide... function New (line 28) | func New() *ReplicateProvider { function readTokenFromStdin (line 102) | func readTokenFromStdin() (string, error) { function readTokenInteractively (line 111) | func readTokenInteractively(registryHost string) (string, error) { function getDisplayTokenURL (line 152) | func getDisplayTokenURL(registryHost string) (string, error) { function addressWithScheme (line 176) | func addressWithScheme(address string) string { function maybeOpenBrowser (line 184) | func maybeOpenBrowser(urlToOpen string) { function checkTokenFormat (line 196) | func checkTokenFormat(token string) error { function verifyToken (line 204) | func verifyToken(registryHost string, token string) (username string, er... FILE: pkg/provider/replicate/replicate_test.go function TestReplicateProvider_Name (line 17) | func TestReplicateProvider_Name(t *testing.T) { function TestReplicateProvider_MatchesRegistry (line 22) | func TestReplicateProvider_MatchesRegistry(t *testing.T) { function TestReplicateProvider_PostPush (line 38) | func TestReplicateProvider_PostPush(t *testing.T) { function TestCheckTokenFormat (line 67) | func TestCheckTokenFormat(t *testing.T) { function TestVerifyToken (line 103) | func TestVerifyToken(t *testing.T) { function TestGetDisplayTokenURL (line 148) | func TestGetDisplayTokenURL(t *testing.T) { function TestAddressWithScheme (line 176) | func TestAddressWithScheme(t *testing.T) { FILE: pkg/provider/setup/setup.go function Init (line 16) | func Init() { FILE: pkg/provider/setup/setup_test.go function TestInit (line 11) | func TestInit(t *testing.T) { FILE: pkg/registry/client.go type Platform (line 11) | type Platform struct type PlatformManifest (line 17) | type PlatformManifest struct type RetryEvent (line 28) | type RetryEvent struct type RetryCallback (line 40) | type RetryCallback type RetryConfig (line 43) | type RetryConfig struct type WriteLayerOptions (line 53) | type WriteLayerOptions struct type Client (line 66) | type Client interface FILE: pkg/registry/client_test.go function TestInspect (line 19) | func TestInspect(t *testing.T) { function TestIsRetryableError (line 84) | func TestIsRetryableError(t *testing.T) { FILE: pkg/registry/config.go constant DefaultChunkSize (line 14) | DefaultChunkSize = 96 * 1024 * 1024 constant DefaultMultipartThreshold (line 20) | DefaultMultipartThreshold = 128 * 1024 * 1024 constant chunkSizeMargin (line 24) | chunkSizeMargin = 64 * 1024 constant envPushDefaultChunkSize (line 29) | envPushDefaultChunkSize = "COG_PUSH_DEFAULT_CHUNK_SIZE" constant envMultipartThreshold (line 32) | envMultipartThreshold = "COG_PUSH_MULTIPART_THRESHOLD" function getDefaultChunkSize (line 37) | func getDefaultChunkSize() int64 { function getMultipartThreshold (line 47) | func getMultipartThreshold() int64 { FILE: pkg/registry/config_test.go function TestGetDefaultChunkSize (line 9) | func TestGetDefaultChunkSize(t *testing.T) { function TestEffectiveChunkSize (line 36) | func TestEffectiveChunkSize(t *testing.T) { function TestGetMultipartThreshold (line 107) | func TestGetMultipartThreshold(t *testing.T) { FILE: pkg/registry/manifest_result.go type ManifestResult (line 5) | type ManifestResult struct method IsIndex (line 17) | func (m *ManifestResult) IsIndex() bool { method IsSinglePlatform (line 21) | func (m *ManifestResult) IsSinglePlatform() bool { FILE: pkg/registry/push_test.go function TestPushOperations (line 16) | func TestPushOperations(t *testing.T) { FILE: pkg/registry/registry_client.go type RegistryClient (line 35) | type RegistryClient struct method remoteOptions (line 51) | func (c *RegistryClient) remoteOptions(ctx context.Context) []remote.O... method Inspect (line 59) | func (c *RegistryClient) Inspect(ctx context.Context, imageRef string,... method GetImage (line 206) | func (c *RegistryClient) GetImage(ctx context.Context, imageRef string... method GetDescriptor (line 277) | func (c *RegistryClient) GetDescriptor(ctx context.Context, imageRef s... method Exists (line 291) | func (c *RegistryClient) Exists(ctx context.Context, imageRef string) ... method PushImage (line 318) | func (c *RegistryClient) PushImage(ctx context.Context, ref string, im... method PushIndex (line 332) | func (c *RegistryClient) PushIndex(ctx context.Context, ref string, id... method WriteLayer (line 444) | func (c *RegistryClient) WriteLayer(ctx context.Context, opts WriteLay... method writeLayerMultipart (line 523) | func (c *RegistryClient) writeLayerMultipart(ctx context.Context, repo... method checkBlobExists (line 583) | func (c *RegistryClient) checkBlobExists(ctx context.Context, client *... method initiateUpload (line 650) | func (c *RegistryClient) initiateUpload(ctx context.Context, client *h... method uploadBlobChunks (line 713) | func (c *RegistryClient) uploadBlobChunks(ctx context.Context, client ... method tryMultipartWithFallback (line 755) | func (c *RegistryClient) tryMultipartWithFallback(ctx context.Context,... method tryMultipartUpload (line 784) | func (c *RegistryClient) tryMultipartUpload(ctx context.Context, clien... method uploadBlobSingle (line 840) | func (c *RegistryClient) uploadBlobSingle(ctx context.Context, client ... method uploadChunk (line 896) | func (c *RegistryClient) uploadChunk(ctx context.Context, client *http... method commitUpload (line 973) | func (c *RegistryClient) commitUpload(ctx context.Context, client *htt... function NewRegistryClient (line 43) | func NewRegistryClient() Client { function checkError (line 301) | func checkError(err error, codes ...transport.ErrorCode) bool { function http1OnlyTransport (line 354) | func http1OnlyTransport() *http.Transport { function tlsConfigHTTP1Only (line 363) | func tlsConfigHTTP1Only(base *tls.Config) *tls.Config { function DefaultRetryBackoff (line 374) | func DefaultRetryBackoff() remote.Backoff { function isRetryableError (line 385) | func isRetryableError(err error) bool { type uploadSession (line 610) | type uploadSession struct method effectiveChunkSize (line 628) | func (s uploadSession) effectiveChunkSize() int64 { type progressReader (line 959) | type progressReader struct method Read (line 964) | func (pr *progressReader) Read(p []byte) (int, error) { function pickDefaultImage (line 1003) | func pickDefaultImage(ref name.Reference, idx *v1.IndexManifest, opts ..... FILE: pkg/registry/registrytest/mock_client.go type MockRegistryClient (line 11) | type MockRegistryClient struct method Exists (line 21) | func (c *MockRegistryClient) Exists(ctx context.Context, imageRef stri... method GetImage (line 26) | func (c *MockRegistryClient) GetImage(ctx context.Context, imageRef st... method Inspect (line 30) | func (c *MockRegistryClient) Inspect(ctx context.Context, imageRef str... method AddMockImage (line 34) | func (c *MockRegistryClient) AddMockImage(imageRef string) { method PushImage (line 38) | func (c *MockRegistryClient) PushImage(ctx context.Context, ref string... method PushIndex (line 43) | func (c *MockRegistryClient) PushIndex(ctx context.Context, ref string... method GetDescriptor (line 48) | func (c *MockRegistryClient) GetDescriptor(ctx context.Context, imageR... method WriteLayer (line 52) | func (c *MockRegistryClient) WriteLayer(ctx context.Context, opts regi... function NewMockRegistryClient (line 15) | func NewMockRegistryClient() *MockRegistryClient { FILE: pkg/registry_testhelpers/registry_container.go function StartTestRegistry (line 36) | func StartTestRegistry(t *testing.T, opts ...Option) *RegistryContainer { function StartTestRegistryWithCleanup (line 51) | func StartTestRegistryWithCleanup(ctx context.Context, opts ...Option) (... type RegistryContainer (line 114) | type RegistryContainer struct method ImageRef (line 119) | func (c *RegistryContainer) ImageRef(ref string) string { method ImageRefForTest (line 123) | func (c *RegistryContainer) ImageRefForTest(t *testing.T, label string... method CloneRepo (line 131) | func (c *RegistryContainer) CloneRepo(t *testing.T, existingRepo, newR... method CloneRepoForTest (line 140) | func (c *RegistryContainer) CloneRepoForTest(t *testing.T, repo string... method ImageExists (line 144) | func (c *RegistryContainer) ImageExists(t *testing.T, ref string) error { method RegistryHost (line 160) | func (c *RegistryContainer) RegistryHost() string { type Option (line 164) | type Option function WithAuth (line 166) | func WithAuth(username, password string) func(*options) { type options (line 175) | type options struct function generateHtpasswd (line 179) | func generateHtpasswd(username, password string) (string, error) { FILE: pkg/requirements/requirements.go constant RequirementsFile (line 15) | RequirementsFile = "requirements.txt" constant OverridesFile (line 16) | OverridesFile = "overrides.txt" function GenerateRequirements (line 18) | func GenerateRequirements(tmpDir string, path string, fileName string) (... function CurrentRequirements (line 33) | func CurrentRequirements(tmpDir string) (string, error) { function ReadRequirements (line 45) | func ReadRequirements(path string) ([]string, error) { function scanLinesWithContinuations (line 87) | func scanLinesWithContinuations(data []byte, atEOF bool) (advance int, t... function SplitPinnedPythonRequirement (line 137) | func SplitPinnedPythonRequirement(requirement string) (name string, vers... function PackageName (line 179) | func PackageName(pipRequirement string) string { function VersionSpecifier (line 188) | func VersionSpecifier(pipRequirement string) string { function Versions (line 198) | func Versions(pipRequirement string) []string { FILE: pkg/requirements/requirements_test.go function TestPythonRequirements (line 13) | func TestPythonRequirements(t *testing.T) { function TestReadRequirements (line 25) | func TestReadRequirements(t *testing.T) { function TestReadRequirementsLineContinuations (line 36) | func TestReadRequirementsLineContinuations(t *testing.T) { function TestReadRequirementsStripComments (line 47) | func TestReadRequirementsStripComments(t *testing.T) { function TestReadRequirementsComplex (line 58) | func TestReadRequirementsComplex(t *testing.T) { function TestReadRequirementsLongLine (line 77) | func TestReadRequirementsLongLine(t *testing.T) { function TestComfyUIRequirements (line 145) | func TestComfyUIRequirements(t *testing.T) { function TestTensorflowRequirements (line 319) | func TestTensorflowRequirements(t *testing.T) { function TestSplitPinnedPythonRequirement (line 389) | func TestSplitPinnedPythonRequirement(t *testing.T) { function TestReadRequirementsWithEditable (line 424) | func TestReadRequirementsWithEditable(t *testing.T) { function TestVersionSpecifier (line 435) | func TestVersionSpecifier(t *testing.T) { function TestPackageName (line 440) | func TestPackageName(t *testing.T) { function TestVersions (line 445) | func TestVersions(t *testing.T) { function checkRequirements (line 450) | func checkRequirements(t *testing.T, expected []string, actual []string) { FILE: pkg/schema/errors.go type SchemaError (line 6) | type SchemaError struct method Error (line 11) | func (e *SchemaError) Error() string { return e.Message } type SchemaErrorKind (line 14) | type SchemaErrorKind constant ErrParse (line 17) | ErrParse SchemaErrorKind = iota constant ErrPredictorNotFound (line 18) | ErrPredictorNotFound constant ErrMethodNotFound (line 19) | ErrMethodNotFound constant ErrMissingReturnType (line 20) | ErrMissingReturnType constant ErrMissingTypeAnnotation (line 21) | ErrMissingTypeAnnotation constant ErrUnsupportedType (line 22) | ErrUnsupportedType constant ErrDefaultFactoryNotSupported (line 23) | ErrDefaultFactoryNotSupported constant ErrInvalidConstraint (line 24) | ErrInvalidConstraint constant ErrInvalidPredictRef (line 25) | ErrInvalidPredictRef constant ErrOptionalOutput (line 26) | ErrOptionalOutput constant ErrConcatIteratorNotStr (line 27) | ErrConcatIteratorNotStr constant ErrChoicesNotResolvable (line 28) | ErrChoicesNotResolvable constant ErrDefaultNotResolvable (line 29) | ErrDefaultNotResolvable constant ErrUnresolvableType (line 30) | ErrUnresolvableType constant ErrOther (line 31) | ErrOther function NewError (line 35) | func NewError(kind SchemaErrorKind, msg string) *SchemaError { function WrapError (line 40) | func WrapError(kind SchemaErrorKind, msg string, inner error) *SchemaErr... function errParse (line 47) | func errParse(msg string) error { //nolint:unused // used by generator.g... function errPredictorNotFound (line 51) | func errPredictorNotFound(name string) error { //nolint:unused // used b... function errMethodNotFound (line 55) | func errMethodNotFound(class, method string) error { //nolint:unused // ... function errMissingReturnType (line 59) | func errMissingReturnType(method string) error { //nolint:unused // used... function errMissingTypeAnnotation (line 63) | func errMissingTypeAnnotation(method, param string) error { //nolint:unu... function errUnsupportedType (line 67) | func errUnsupportedType(msg string) error { function errDefaultFactoryNotSupported (line 71) | func errDefaultFactoryNotSupported(param string) error { //nolint:unused... function errInvalidPredictRef (line 78) | func errInvalidPredictRef(ref string) error { function errOptionalOutput (line 85) | func errOptionalOutput() error { function errConcatIteratorNotStr (line 89) | func errConcatIteratorNotStr(got string) error { function errChoicesNotResolvable (line 93) | func errChoicesNotResolvable(param string) error { //nolint:unused // us... function errUnresolvableImportedType (line 100) | func errUnresolvableImportedType(name, module string) error { function errUnresolvableType (line 111) | func errUnresolvableType(name string) error { function errDefaultNotResolvable (line 122) | func errDefaultNotResolvable(param, expr string) error { //nolint:unused... FILE: pkg/schema/generator.go type Parser (line 16) | type Parser function Generate (line 28) | func Generate(predictRef string, sourceDir string, mode Mode, parse Pars... function GenerateFromSource (line 62) | func GenerateFromSource(source []byte, predictRef string, mode Mode, par... function GenerateCombined (line 76) | func GenerateCombined(sourceDir string, predictRef string, trainRef stri... function MergeSchemas (line 123) | func MergeSchemas(predict, train map[string]any) map[string]any { function parsePredictRef (line 155) | func parsePredictRef(ref string) (filePath string, name string, err erro... FILE: pkg/schema/generator_test.go function mockParser (line 14) | func mockParser(source []byte, predictRef string, mode Mode, sourceDir s... function failParser (line 29) | func failParser(_ []byte, _ string, _ Mode, _ string) (*PredictorInfo, e... function TestParsePredictRef (line 37) | func TestParsePredictRef(t *testing.T) { function TestGenerateFromSource (line 73) | func TestGenerateFromSource(t *testing.T) { function TestGenerateFromSourceTrainMode (line 85) | func TestGenerateFromSourceTrainMode(t *testing.T) { function TestGenerateFromSourceParseError (line 96) | func TestGenerateFromSourceParseError(t *testing.T) { function TestGenerateReadsFile (line 106) | func TestGenerateReadsFile(t *testing.T) { function TestGenerateMissingFile (line 119) | func TestGenerateMissingFile(t *testing.T) { function TestGenerateInvalidRef (line 126) | func TestGenerateInvalidRef(t *testing.T) { function TestGenerateCogOpenAPISchemaEnv (line 138) | func TestGenerateCogOpenAPISchemaEnv(t *testing.T) { function TestGenerateCogOpenAPISchemaEnvMissingFile (line 155) | func TestGenerateCogOpenAPISchemaEnvMissingFile(t *testing.T) { function TestGenerateCogOpenAPISchemaEnvNotSet (line 164) | func TestGenerateCogOpenAPISchemaEnvNotSet(t *testing.T) { FILE: pkg/schema/openapi.go function GenerateOpenAPISchema (line 11) | func GenerateOpenAPISchema(info *PredictorInfo) ([]byte, error) { function buildOpenAPISpec (line 22) | func buildOpenAPISpec(info *PredictorInfo) map[string]any { type enumSchema (line 265) | type enumSchema struct function buildInputSchema (line 271) | func buildInputSchema(info *PredictorInfo) (map[string]any, []enumSchema) { function removeTitleNextToRef (line 390) | func removeTitleNextToRef(v any) { function fixNullableAnyOf (line 415) | func fixNullableAnyOf(v any) { type orderedMapAny (line 491) | type orderedMapAny struct method Set (line 500) | func (m *orderedMapAny) Set(key string, value any) { method Get (line 507) | func (m *orderedMapAny) Get(key string) (any, bool) { method Delete (line 512) | func (m *orderedMapAny) Delete(key string) { method Entries (line 525) | func (m *orderedMapAny) Entries(fn func(key string, value any)) { method MarshalJSON (line 532) | func (m *orderedMapAny) MarshalJSON() ([]byte, error) { function newOrderedMapAny (line 496) | func newOrderedMapAny() *orderedMapAny { FILE: pkg/schema/openapi_test.go function simplePredictor (line 15) | func simplePredictor() *PredictorInfo { function ptr (line 30) | func ptr[T any](v T) *T { return &v } function parseSpec (line 34) | func parseSpec(t *testing.T, info *PredictorInfo) map[string]any { function getPath (line 43) | func getPath(m map[string]any, keys ...string) any { function TestGeneratesValidOpenAPI (line 59) | func TestGeneratesValidOpenAPI(t *testing.T) { function TestPredictEndpoints (line 67) | func TestPredictEndpoints(t *testing.T) { function TestTrainEndpoints (line 84) | func TestTrainEndpoints(t *testing.T) { function TestFixedComponentSchemas (line 110) | func TestFixedComponentSchemas(t *testing.T) { function TestInputRequiredField (line 144) | func TestInputRequiredField(t *testing.T) { function TestInputOptionalFieldNotRequired (line 152) | func TestInputOptionalFieldNotRequired(t *testing.T) { function TestInputDefaultValue (line 179) | func TestInputDefaultValue(t *testing.T) { function TestInputDescription (line 201) | func TestInputDescription(t *testing.T) { function TestInputNumericConstraints (line 222) | func TestInputNumericConstraints(t *testing.T) { function TestInputStringConstraints (line 245) | func TestInputStringConstraints(t *testing.T) { function TestInputDeprecated (line 270) | func TestInputDeprecated(t *testing.T) { function TestInputXOrder (line 291) | func TestInputXOrder(t *testing.T) { function TestInputRepeatedType (line 316) | func TestInputRepeatedType(t *testing.T) { function TestChoicesGenerateEnum (line 342) | func TestChoicesGenerateEnum(t *testing.T) { function TestIntegerChoices (line 378) | func TestIntegerChoices(t *testing.T) { function TestOutputSingle (line 409) | func TestOutputSingle(t *testing.T) { function TestOutputList (line 416) | func TestOutputList(t *testing.T) { function TestOutputIterator (line 432) | func TestOutputIterator(t *testing.T) { function TestOutputConcatenateIterator (line 446) | func TestOutputConcatenateIterator(t *testing.T) { function TestOutputObject (line 461) | func TestOutputObject(t *testing.T) { function TestOutputPath (line 509) | func TestOutputPath(t *testing.T) { function TestRemoveTitleNextToRef (line 527) | func TestRemoveTitleNextToRef(t *testing.T) { function TestRemoveTitleNextToRefNested (line 537) | func TestRemoveTitleNextToRefNested(t *testing.T) { function TestFixNullableAnyOf (line 551) | func TestFixNullableAnyOf(t *testing.T) { function TestFixNullableAnyOfNoOp (line 564) | func TestFixNullableAnyOfNoOp(t *testing.T) { function TestTitleCaseWords (line 581) | func TestTitleCaseWords(t *testing.T) { function TestTitleCaseSingleWord (line 587) | func TestTitleCaseSingleWord(t *testing.T) { function TestOutputIsValidJSON (line 597) | func TestOutputIsValidJSON(t *testing.T) { function TestMultipleInputTypes (line 610) | func TestMultipleInputTypes(t *testing.T) { function TestNoInputs (line 689) | func TestNoInputs(t *testing.T) { function TestOutputObjectNoFields (line 704) | func TestOutputObjectNoFields(t *testing.T) { function TestOrderedMapAnyJSON (line 721) | func TestOrderedMapAnyJSON(t *testing.T) { function TestOrderedMapAnyDelete (line 732) | func TestOrderedMapAnyDelete(t *testing.T) { FILE: pkg/schema/python/parser.go function ParsePredictor (line 29) | func ParsePredictor(source []byte, predictRef string, mode schema.Mode, ... function namedChildren (line 105) | func namedChildren(n *sitter.Node) []*sitter.Node { function allChildren (line 115) | func allChildren(n *sitter.Node) []*sitter.Node { function content (line 125) | func content(n *sitter.Node, source []byte) string { function collectImports (line 133) | func collectImports(root *sitter.Node, source []byte) *schema.ImportCont... function parseImportFrom (line 155) | func parseImportFrom(node *sitter.Node, source []byte, ctx *schema.Impor... type moduleScope (line 212) | type moduleScope function collectModuleScope (line 214) | func collectModuleScope(root *sitter.Node, source []byte) moduleScope { function resolveDefaultExpr (line 251) | func resolveDefaultExpr(node *sitter.Node, source []byte, scope moduleSc... function resolveChoicesExpr (line 265) | func resolveChoicesExpr(node *sitter.Node, source []byte, scope moduleSc... function resolveChoicesCall (line 315) | func resolveChoicesCall(node *sitter.Node, source []byte, scope moduleSc... function collectModelClasses (line 368) | func collectModelClasses(root *sitter.Node, source []byte, imports *sche... function resolveExternalModels (line 407) | func resolveExternalModels(sourceDir string, imports *schema.ImportConte... function moduleToFilePath (line 483) | func moduleToFilePath(module string) string { function isKnownExternalModule (line 496) | func isKnownExternalModule(module string) bool { function unwrapClass (line 521) | func unwrapClass(node *sitter.Node) *sitter.Node { function unwrapFunction (line 535) | func unwrapFunction(node *sitter.Node) *sitter.Node { function inheritsFromBaseModel (line 549) | func inheritsFromBaseModel(classNode *sitter.Node, source []byte, import... function extractClassAnnotations (line 572) | func extractClassAnnotations(classNode *sitter.Node, source []byte) []sc... function parseAnnotatedAssignment (line 599) | func parseAnnotatedAssignment(node *sitter.Node, source []byte) (schema.... function parseBareAnnotation (line 622) | func parseBareAnnotation(node *sitter.Node, source []byte) (schema.Model... function parseTypeFromString (line 643) | func parseTypeFromString(s string) (schema.TypeAnnotation, bool) { function splitTopLevelCommas (line 699) | func splitTopLevelCommas(s string) []string { type inputCallInfo (line 724) | type inputCallInfo struct type inputMethodInfo (line 736) | type inputMethodInfo struct type inputRegistry (line 741) | type inputRegistry struct function newInputRegistry (line 746) | func newInputRegistry() *inputRegistry { function collectInputRegistry (line 753) | func collectInputRegistry(root *sitter.Node, source []byte, imports *sch... function collectInputAttribute (line 791) | func collectInputAttribute(className string, assignment *sitter.Node, so... function collectInputMethod (line 811) | func collectInputMethod(className string, funcNode *sitter.Node, source ... function findReturnInputCall (line 871) | func findReturnInputCall(body *sitter.Node, source []byte, imports *sche... function resolveInputReference (line 885) | func resolveInputReference(node *sitter.Node, source []byte, registry *i... function findTargetFunction (line 957) | func findTargetFunction(root *sitter.Node, source []byte, predictRef, me... function findMethodInClass (line 988) | func findMethodInClass(classNode *sitter.Node, source []byte, className,... function firstParamIsSelf (line 1012) | func firstParamIsSelf(params *sitter.Node, source []byte) bool { function extractInputs (line 1021) | func extractInputs( function parseTypedParameter (line 1074) | func parseTypedParameter(node *sitter.Node, source []byte, order int, me... function parseTypedDefaultParameter (line 1113) | func parseTypedDefaultParameter( function parseTypeAnnotation (line 1213) | func parseTypeAnnotation(node *sitter.Node, source []byte) (schema.TypeA... function errUnsupported (line 1316) | func errUnsupported(msg string) error { function isInputCall (line 1324) | func isInputCall(node *sitter.Node, source []byte, imports *schema.Impor... function parseInputCall (line 1342) | func parseInputCall(node *sitter.Node, source []byte, paramName string, ... function parseDefaultValue (line 1419) | func parseDefaultValue(node *sitter.Node, source []byte) (schema.Default... function parseStringLiteral (line 1486) | func parseStringLiteral(node *sitter.Node, source []byte) (string, bool) { function parseNumberLiteral (line 1509) | func parseNumberLiteral(node *sitter.Node, source []byte) (float64, bool) { function parseBoolLiteral (line 1518) | func parseBoolLiteral(node *sitter.Node, source []byte) (bool, bool) { function parseListLiteral (line 1535) | func parseListLiteral(node *sitter.Node, source []byte) ([]schema.Defaul... function parseDictLiteral (line 1550) | func parseDictLiteral(node *sitter.Node, source []byte) ([]schema.Defaul... function parseSetLiteral (line 1573) | func parseSetLiteral(node *sitter.Node, source []byte) ([]schema.Default... FILE: pkg/schema/python/parser_fuzz_test.go function FuzzParsePredictor (line 11) | func FuzzParsePredictor(f *testing.F) { function FuzzParseTypeAnnotation (line 75) | func FuzzParseTypeAnnotation(f *testing.F) { FILE: pkg/schema/python/parser_test.go function parse (line 15) | func parse(t *testing.T, source, predictRef string) *schema.PredictorInfo { function parseErr (line 23) | func parseErr(t *testing.T, source, predictRef string, mode schema.Mode)... function TestSimpleStringPredictor (line 36) | func TestSimpleStringPredictor(t *testing.T) { function TestMultipleInputsWithDefaults (line 58) | func TestMultipleInputsWithDefaults(t *testing.T) { function TestOptionalInputPipeNone (line 98) | func TestOptionalInputPipeNone(t *testing.T) { function TestOptionalInputTyping (line 118) | func TestOptionalInputTyping(t *testing.T) { function TestListInput (line 141) | func TestListInput(t *testing.T) { function TestChoicesLiteralList (line 160) | func TestChoicesLiteralList(t *testing.T) { function TestChoicesModuleLevelListVar (line 177) | func TestChoicesModuleLevelListVar(t *testing.T) { function TestChoicesListDictKeys (line 196) | func TestChoicesListDictKeys(t *testing.T) { function TestChoicesListDictValues (line 219) | func TestChoicesListDictValues(t *testing.T) { function TestChoicesDictKeysPlusLiteral (line 237) | func TestChoicesDictKeysPlusLiteral(t *testing.T) { function TestChoicesIntegerDictKeys (line 256) | func TestChoicesIntegerDictKeys(t *testing.T) { function TestChoicesConcatTwoVars (line 276) | func TestChoicesConcatTwoVars(t *testing.T) { function TestChoicesVarNotAListErrors (line 300) | func TestChoicesVarNotAListErrors(t *testing.T) { function TestChoicesUndefinedVarErrors (line 314) | func TestChoicesUndefinedVarErrors(t *testing.T) { function TestChoicesArbitraryCallErrors (line 326) | func TestChoicesArbitraryCallErrors(t *testing.T) { function TestChoicesListComprehensionErrors (line 338) | func TestChoicesListComprehensionErrors(t *testing.T) { function TestChoicesErrorIncludesParamName (line 350) | func TestChoicesErrorIncludesParamName(t *testing.T) { function TestChoicesNestedVarNotInScope (line 362) | func TestChoicesNestedVarNotInScope(t *testing.T) { function TestStandaloneFunction (line 381) | func TestStandaloneFunction(t *testing.T) { function TestIteratorOutput (line 402) | func TestIteratorOutput(t *testing.T) { function TestConcatenateIteratorOutput (line 419) | func TestConcatenateIteratorOutput(t *testing.T) { function TestConcatenateIteratorNotStrErrors (line 434) | func TestConcatenateIteratorNotStrErrors(t *testing.T) { function TestListOutput (line 446) | func TestListOutput(t *testing.T) { function TestBaseModelOutput (line 461) | func TestBaseModelOutput(t *testing.T) { function TestOptionalOutputErrors (line 489) | func TestOptionalOutputErrors(t *testing.T) { function TestOptionalOutputPipeNoneErrors (line 502) | func TestOptionalOutputPipeNoneErrors(t *testing.T) { function TestTrainMode (line 518) | func TestTrainMode(t *testing.T) { function TestNonBasePredictor (line 535) | func TestNonBasePredictor(t *testing.T) { function TestDefaultFactoryError (line 555) | func TestDefaultFactoryError(t *testing.T) { function TestDefaultModuleLevelStringInInput (line 571) | func TestDefaultModuleLevelStringInInput(t *testing.T) { function TestDefaultModuleLevelIntInInput (line 589) | func TestDefaultModuleLevelIntInInput(t *testing.T) { function TestDefaultModuleLevelListInInput (line 607) | func TestDefaultModuleLevelListInInput(t *testing.T) { function TestDefaultModuleLevelVarPlain (line 627) | func TestDefaultModuleLevelVarPlain(t *testing.T) { function TestDefaultUndefinedVarPlainErrors (line 645) | func TestDefaultUndefinedVarPlainErrors(t *testing.T) { function TestInputRegistryAttribute (line 661) | func TestInputRegistryAttribute(t *testing.T) { function TestInputRegistryMethod (line 692) | func TestInputRegistryMethod(t *testing.T) { function TestPredictorNotFound (line 725) | func TestPredictorNotFound(t *testing.T) { function TestMethodNotFound (line 737) | func TestMethodNotFound(t *testing.T) { function TestMissingReturnType (line 749) | func TestMissingReturnType(t *testing.T) { function TestMissingTypeAnnotation (line 761) | func TestMissingTypeAnnotation(t *testing.T) { function TestAllPrimitiveInputTypes (line 777) | func TestAllPrimitiveInputTypes(t *testing.T) { function TestInputConstraints (line 812) | func TestInputConstraints(t *testing.T) { function TestNegativeNumberDefault (line 845) | func TestNegativeNumberDefault(t *testing.T) { function TestBoolDefault (line 861) | func TestBoolDefault(t *testing.T) { function TestParameterOrdering (line 881) | func TestParameterOrdering(t *testing.T) { function TestDeprecatedInput (line 910) | func TestDeprecatedInput(t *testing.T) { function TestFileType (line 929) | func TestFileType(t *testing.T) { function TestSecretType (line 947) | func TestSecretType(t *testing.T) { function TestMultipleClassesFindsTarget (line 965) | func TestMultipleClassesFindsTarget(t *testing.T) { function TestBaseModelOutputWithDefaults (line 988) | func TestBaseModelOutputWithDefaults(t *testing.T) { function TestPydanticBaseModelOutput (line 1014) | func TestPydanticBaseModelOutput(t *testing.T) { function TestPydanticBaseModelDottedOutput (line 1044) | func TestPydanticBaseModelDottedOutput(t *testing.T) { function TestPydanticBaseModelDirectImport (line 1065) | func TestPydanticBaseModelDirectImport(t *testing.T) { function TestDictOutput (line 1090) | func TestDictOutput(t *testing.T) { function TestParameterizedDictOutput (line 1102) | func TestParameterizedDictOutput(t *testing.T) { function TestBareListOutput (line 1120) | func TestBareListOutput(t *testing.T) { function TestNoInputPredictor (line 1138) | func TestNoInputPredictor(t *testing.T) { function TestDefaultFalse (line 1155) | func TestDefaultFalse(t *testing.T) { function TestDefaultZeroInt (line 1172) | func TestDefaultZeroInt(t *testing.T) { function TestDefaultZeroFloat (line 1189) | func TestDefaultZeroFloat(t *testing.T) { function TestDefaultEmptyString (line 1206) | func TestDefaultEmptyString(t *testing.T) { function TestDefaultNegativeInt (line 1223) | func TestDefaultNegativeInt(t *testing.T) { function TestAsyncIteratorOutput (line 1243) | func TestAsyncIteratorOutput(t *testing.T) { function TestAsyncConcatenateIteratorOutput (line 1259) | func TestAsyncConcatenateIteratorOutput(t *testing.T) { function TestTypingListCapitalL (line 1279) | func TestTypingListCapitalL(t *testing.T) { function TestTypingUnionStrNone (line 1295) | func TestTypingUnionStrNone(t *testing.T) { function TestAllOptionalInputs (line 1316) | func TestAllOptionalInputs(t *testing.T) { function TestListPathInput (line 1340) | func TestListPathInput(t *testing.T) { function TestDictStrStrOutput (line 1359) | func TestDictStrStrOutput(t *testing.T) { function TestDictStrIntOutput (line 1374) | func TestDictStrIntOutput(t *testing.T) { function TestNestedDictOutput (line 1389) | func TestNestedDictOutput(t *testing.T) { function TestDictOfListOutput (line 1407) | func TestDictOfListOutput(t *testing.T) { function TestListOfDictOutput (line 1425) | func TestListOfDictOutput(t *testing.T) { function TestListOfListOutput (line 1443) | func TestListOfListOutput(t *testing.T) { function TestTripleNestedDictOutput (line 1461) | func TestTripleNestedDictOutput(t *testing.T) { function TestListOfDictOfListOutput (line 1487) | func TestListOfDictOfListOutput(t *testing.T) { function TestIteratorOfDictOutput (line 1513) | func TestIteratorOfDictOutput(t *testing.T) { function TestIteratorOfListOutput (line 1532) | func TestIteratorOfListOutput(t *testing.T) { function TestDictOfPathOutput (line 1551) | func TestDictOfPathOutput(t *testing.T) { function TestListOfPathOutput (line 1567) | func TestListOfPathOutput(t *testing.T) { function TestUnresolvableImportedTypeError (line 1587) | func TestUnresolvableImportedTypeError(t *testing.T) { function TestUnresolvableUndefinedTypeError (line 1603) | func TestUnresolvableUndefinedTypeError(t *testing.T) { function TestUnresolvableDottedImportTypeError (line 1618) | func TestUnresolvableDottedImportTypeError(t *testing.T) { function TestUnresolvableTypeTorchTensor (line 1633) | func TestUnresolvableTypeTorchTensor(t *testing.T) { function TestUnresolvableTypeNumpyArray (line 1648) | func TestUnresolvableTypeNumpyArray(t *testing.T) { function TestDictWithUnresolvableValueTypeErrors (line 1663) | func TestDictWithUnresolvableValueTypeErrors(t *testing.T) { function TestModelFieldDictWithUnresolvableValueTypeErrors (line 1679) | func TestModelFieldDictWithUnresolvableValueTypeErrors(t *testing.T) { function TestPydanticV1CompatOutput (line 1702) | func TestPydanticV1CompatOutput(t *testing.T) { function TestPydanticOutputWithOptionalField (line 1731) | func TestPydanticOutputWithOptionalField(t *testing.T) { function TestPydanticOutputDefaultedFieldNotNullable (line 1758) | func TestPydanticOutputDefaultedFieldNotNullable(t *testing.T) { function TestPydanticOutputOptionalFieldNullable (line 1805) | func TestPydanticOutputOptionalFieldNullable(t *testing.T) { function TestPydanticOutputWithListField (line 1835) | func TestPydanticOutputWithListField(t *testing.T) { function TestPydanticOutputWithDictField (line 1859) | func TestPydanticOutputWithDictField(t *testing.T) { function TestPydanticOutputWithOptionalDictField (line 1894) | func TestPydanticOutputWithOptionalDictField(t *testing.T) { function writeFile (line 1924) | func writeFile(t *testing.T, dir, name, content string) { function parseFile (line 1932) | func parseFile(t *testing.T, dir, filename, predictRef string) *schema.P... function TestCrossFileBaseModelSameDir (line 1941) | func TestCrossFileBaseModelSameDir(t *testing.T) { function TestCrossFileRelativeImport (line 1973) | func TestCrossFileRelativeImport(t *testing.T) { function TestCrossFileSubpackageImport (line 2001) | func TestCrossFileSubpackageImport(t *testing.T) { function TestCrossFileRelativeSubpackage (line 2033) | func TestCrossFileRelativeSubpackage(t *testing.T) { function TestCrossFileMultipleModelsFromSameFile (line 2060) | func TestCrossFileMultipleModelsFromSameFile(t *testing.T) { function TestCrossFileWithOptionalField (line 2096) | func TestCrossFileWithOptionalField(t *testing.T) { function TestCrossFileAliasedImport (line 2135) | func TestCrossFileAliasedImport(t *testing.T) { function TestCrossFileExternalPackageStillErrors (line 2162) | func TestCrossFileExternalPackageStillErrors(t *testing.T) { function TestCrossFileLocalPrecedesExternal (line 2184) | func TestCrossFileLocalPrecedesExternal(t *testing.T) { function TestCrossFileListFieldInExternalModel (line 2212) | func TestCrossFileListFieldInExternalModel(t *testing.T) { function TestCrossFileEndToEndSchemaGeneration (line 2240) | func TestCrossFileEndToEndSchemaGeneration(t *testing.T) { FILE: pkg/schema/schema_type.go type SchemaType (line 12) | type SchemaType struct method JSONSchema (line 66) | func (s SchemaType) JSONSchema() map[string]any { method jsonSchema (line 70) | func (s SchemaType) jsonSchema(topLevel bool) map[string]any { method coreSchema (line 84) | func (s SchemaType) coreSchema() map[string]any { type SchemaTypeKind (line 38) | type SchemaTypeKind constant SchemaPrimitive (line 42) | SchemaPrimitive SchemaTypeKind = iota constant SchemaAny (line 44) | SchemaAny constant SchemaArray (line 46) | SchemaArray constant SchemaDict (line 48) | SchemaDict constant SchemaObject (line 50) | SchemaObject constant SchemaIterator (line 52) | SchemaIterator constant SchemaConcatIterator (line 54) | SchemaConcatIterator type SchemaField (line 58) | type SchemaField struct function SchemaPrim (line 168) | func SchemaPrim(p PrimitiveType) SchemaType { function SchemaAnyType (line 173) | func SchemaAnyType() SchemaType { function SchemaArrayOf (line 178) | func SchemaArrayOf(elem SchemaType) SchemaType { function SchemaDictOf (line 183) | func SchemaDictOf(value SchemaType) SchemaType { function SchemaIteratorOf (line 189) | func SchemaIteratorOf(elem SchemaType) SchemaType { function SchemaConcatIteratorOf (line 194) | func SchemaConcatIteratorOf() SchemaType { function SchemaObjectOf (line 200) | func SchemaObjectOf(fields *OrderedMap[string, SchemaField]) SchemaType { function ResolveSchemaType (line 215) | func ResolveSchemaType(ann TypeAnnotation, ctx *ImportContext, models Mo... function resolveSimpleSchemaType (line 227) | func resolveSimpleSchemaType(ann TypeAnnotation, ctx *ImportContext, mod... function resolveGenericSchemaType (line 254) | func resolveGenericSchemaType(ann TypeAnnotation, ctx *ImportContext, mo... function resolveUnionSchemaType (line 329) | func resolveUnionSchemaType(ann TypeAnnotation) (SchemaType, error) { function resolveModelToSchemaType (line 340) | func resolveModelToSchemaType(modelFields []ModelField, ctx *ImportConte... function resolveFieldSchemaType (line 362) | func resolveFieldSchemaType(ann TypeAnnotation, ctx *ImportContext, mode... FILE: pkg/schema/schema_type_fuzz_test.go function FuzzResolveSchemaType (line 9) | func FuzzResolveSchemaType(f *testing.F) { function FuzzJSONSchema (line 65) | func FuzzJSONSchema(f *testing.F) { constant maxFuzzDepth (line 84) | maxFuzzDepth = 8 function encodeAnnotation (line 87) | func encodeAnnotation(ann TypeAnnotation) []byte { function decodeAnnotation (line 98) | func decodeAnnotation(data []byte, offset int, depth int) (TypeAnnotatio... function decodeSchemaType (line 136) | func decodeSchemaType(data []byte, offset int, depth int) (SchemaType, i... FILE: pkg/schema/types.go type Mode (line 10) | type Mode constant ModePredict (line 13) | ModePredict Mode = iota constant ModeTrain (line 14) | ModeTrain type PrimitiveType (line 18) | type PrimitiveType method JSONType (line 32) | func (p PrimitiveType) JSONType() map[string]any { method String (line 53) | func (p PrimitiveType) String() string { constant TypeBool (line 21) | TypeBool PrimitiveType = iota constant TypeFloat (line 22) | TypeFloat constant TypeInteger (line 23) | TypeInteger constant TypeString (line 24) | TypeString constant TypePath (line 25) | TypePath constant TypeFile (line 26) | TypeFile constant TypeSecret (line 27) | TypeSecret constant TypeAny (line 28) | TypeAny function PrimitiveFromName (line 62) | func PrimitiveFromName(name string) (PrimitiveType, bool) { type Repetition (line 86) | type Repetition constant Required (line 89) | Required Repetition = iota constant Optional (line 90) | Optional constant Repeated (line 91) | Repeated type FieldType (line 95) | type FieldType struct method JSONType (line 101) | func (ft FieldType) JSONType() map[string]any { type DefaultValue (line 112) | type DefaultValue struct method ToJSON (line 138) | func (d DefaultValue) ToJSON() any { method MarshalJSON (line 172) | func (d DefaultValue) MarshalJSON() ([]byte, error) { type DefaultKind (line 124) | type DefaultKind constant DefaultNone (line 127) | DefaultNone DefaultKind = iota constant DefaultBool (line 128) | DefaultBool constant DefaultInt (line 129) | DefaultInt constant DefaultFloat (line 130) | DefaultFloat constant DefaultString (line 131) | DefaultString constant DefaultList (line 132) | DefaultList constant DefaultDict (line 133) | DefaultDict constant DefaultSet (line 134) | DefaultSet type InputField (line 177) | type InputField struct method IsRequired (line 193) | func (f *InputField) IsRequired() bool { type PredictorInfo (line 198) | type PredictorInfo struct type TypeAnnotation (line 205) | type TypeAnnotation struct type TypeAnnotationKind (line 212) | type TypeAnnotationKind constant TypeAnnotSimple (line 215) | TypeAnnotSimple TypeAnnotationKind = iota constant TypeAnnotGeneric (line 216) | TypeAnnotGeneric constant TypeAnnotUnion (line 217) | TypeAnnotUnion type ImportContext (line 221) | type ImportContext struct method IsCogType (line 238) | func (ctx *ImportContext) IsCogType(name string) bool { method IsTypingType (line 246) | func (ctx *ImportContext) IsTypingType(name string) bool { method IsBaseModel (line 254) | func (ctx *ImportContext) IsBaseModel(name string) bool { method IsBasePredictor (line 262) | func (ctx *ImportContext) IsBasePredictor(name string) bool { type ImportEntry (line 227) | type ImportEntry struct function NewImportContext (line 233) | func NewImportContext() *ImportContext { function ResolveFieldType (line 270) | func ResolveFieldType(ann TypeAnnotation, ctx *ImportContext) (FieldType... function UnwrapOptional (line 325) | func UnwrapOptional(ann TypeAnnotation) (TypeAnnotation, bool) { type ModelField (line 346) | type ModelField struct function TitleCase (line 353) | func TitleCase(s string) string { function TitleCaseSingle (line 364) | func TitleCaseSingle(s string) string { type OrderedMap (line 372) | type OrderedMap struct function NewOrderedMap (line 378) | func NewOrderedMap[K comparable, V any]() *OrderedMap[K, V] { method Set (line 383) | func (m *OrderedMap[K, V]) Set(key K, value V) { method Get (line 391) | func (m *OrderedMap[K, V]) Get(key K) (V, bool) { method Keys (line 397) | func (m *OrderedMap[K, V]) Keys() []K { method Len (line 402) | func (m *OrderedMap[K, V]) Len() int { method Entries (line 407) | func (m *OrderedMap[K, V]) Entries(fn func(key K, value V)) { FILE: pkg/update/state.go type state (line 15) | type state struct function loadState (line 22) | func loadState() (*state, error) { function writeState (line 52) | func writeState(s *state) error { function userDir (line 74) | func userDir() (string, error) { function statePath (line 78) | func statePath() (string, error) { FILE: pkg/update/update.go function isUpdateEnabled (line 17) | func isUpdateEnabled() bool { function DisplayAndCheckForRelease (line 24) | func DisplayAndCheckForRelease(ctx context.Context) error { function startCheckingForRelease (line 49) | func startCheckingForRelease(ctx context.Context) { type updateCheckResponse (line 72) | type updateCheckResponse struct function checkForRelease (line 76) | func checkForRelease(ctx context.Context) (*updateCheckResponse, error) { FILE: pkg/util/console/console.go function ShouldUseColor (line 26) | func ShouldUseColor() bool { type Style (line 44) | type Style constant StyleDefault (line 48) | StyleDefault Style = iota constant StyleSuccess (line 50) | StyleSuccess type Console (line 58) | type Console struct method Debug (line 66) | func (c *Console) Debug(msg string) { method Info (line 71) | func (c *Console) Info(msg string) { method Success (line 77) | func (c *Console) Success(msg string) { method Warn (line 82) | func (c *Console) Warn(msg string) { method Error (line 87) | func (c *Console) Error(msg string) { method Fatal (line 92) | func (c *Console) Fatal(msg string) { method Debugf (line 98) | func (c *Console) Debugf(msg string, v ...any) { method Infof (line 103) | func (c *Console) Infof(msg string, v ...any) { method Successf (line 108) | func (c *Console) Successf(msg string, v ...any) { method Warnf (line 113) | func (c *Console) Warnf(msg string, v ...any) { method Errorf (line 118) | func (c *Console) Errorf(msg string, v ...any) { method Fatalf (line 123) | func (c *Console) Fatalf(msg string, v ...any) { method InfoUnformatted (line 131) | func (c *Console) InfoUnformatted(msg string) { method InfoUnformattedf (line 154) | func (c *Console) InfoUnformattedf(msg string, v ...any) { method Output (line 160) | func (c *Console) Output(s string) { method Bold (line 168) | func (c *Console) Bold(s string) string { method log (line 175) | func (c *Console) log(level Level, msg string) { method logStyled (line 179) | func (c *Console) logStyled(level Level, style Style, msg string) { function stderrTerminalWidth (line 240) | func stderrTerminalWidth() int { function wrapLine (line 258) | func wrapLine(line string, maxWidth int) []string { function visibleWidth (line 288) | func visibleWidth(s string) int { function findCutPoint (line 311) | func findCutPoint(s string, maxWidth int) int { FILE: pkg/util/console/formatting.go function FormatTime (line 9) | func FormatTime(t time.Time) string { FILE: pkg/util/console/global.go function SetLevel (line 17) | func SetLevel(level Level) { function SetColor (line 22) | func SetColor(color bool) { function Debug (line 27) | func Debug(msg string) { function Info (line 32) | func Info(msg string) { function Success (line 37) | func Success(msg string) { function Warn (line 42) | func Warn(msg string) { function Error (line 47) | func Error(msg string) { function Fatal (line 52) | func Fatal(msg string) { function Debugf (line 57) | func Debugf(msg string, v ...any) { function Infof (line 62) | func Infof(msg string, v ...any) { function Successf (line 67) | func Successf(msg string, v ...any) { function Warnf (line 72) | func Warnf(msg string, v ...any) { function Errorf (line 77) | func Errorf(msg string, v ...any) { function Fatalf (line 82) | func Fatalf(msg string, v ...any) { function InfoUnformatted (line 87) | func InfoUnformatted(msg string) { function InfoUnformattedf (line 92) | func InfoUnformattedf(msg string, v ...any) { function Output (line 97) | func Output(s string) { function Bold (line 102) | func Bold(s string) string { function IsTTY (line 107) | func IsTTY(f *os.File) bool { FILE: pkg/util/console/interactive.go type Interactive (line 12) | type Interactive struct method Read (line 19) | func (i Interactive) Read() (string, error) { type InteractiveBool (line 76) | type InteractiveBool struct method Read (line 83) | func (i InteractiveBool) Read() (bool, error) { FILE: pkg/util/console/levels.go type Level (line 14) | type Level method String (line 44) | func (l Level) String() string { constant InvalidLevel (line 18) | InvalidLevel Level = iota - 1 constant DebugLevel (line 19) | DebugLevel constant InfoLevel (line 20) | InfoLevel constant WarnLevel (line 21) | WarnLevel constant ErrorLevel (line 22) | ErrorLevel constant FatalLevel (line 23) | FatalLevel function ParseLevel (line 49) | func ParseLevel(s string) (Level, error) { function MustParseLevel (line 59) | func MustParseLevel(s string) Level { FILE: pkg/util/console/term.go function IsTerminal (line 10) | func IsTerminal() bool { function GetWidth (line 17) | func GetWidth() (uint16, error) { FILE: pkg/util/env.go function GetEnvOrDefault (line 11) | func GetEnvOrDefault[T any](key string, defaultVal T, conversionFunc fun... FILE: pkg/util/errors.go function WrapError (line 7) | func WrapError(err error, message string) error { FILE: pkg/util/files/files.go function Exists (line 23) | func Exists(path string) (bool, error) { function IsEmpty (line 33) | func IsEmpty(path string) (bool, error) { function IsDir (line 44) | func IsDir(path string) (bool, error) { function IsExecutable (line 52) | func IsExecutable(path string) bool { function CopyFile (line 56) | func CopyFile(src string, dest string) error { function WriteIfDifferent (line 76) | func WriteIfDifferent(file, content string) error { function WriteDataURLToFile (line 97) | func WriteDataURLToFile(url string, destination string) (string, error) { function WriteFile (line 139) | func WriteFile(output []byte, outputPath string) (string, error) { FILE: pkg/util/files/files_test.go function TestIsExecutable (line 11) | func TestIsExecutable(t *testing.T) { function TestWriteBadlyFormattedBase64DataURI (line 22) | func TestWriteBadlyFormattedBase64DataURI(t *testing.T) { function TestWriteNotRecognisedBase64DataURL (line 29) | func TestWriteNotRecognisedBase64DataURL(t *testing.T) { FILE: pkg/util/hash.go function SHA256HashFile (line 17) | func SHA256HashFile(path string) (string, error) { function SHA256HashFileWithSaltAndRange (line 33) | func SHA256HashFileWithSaltAndRange(path string, start int, end int, sal... FILE: pkg/util/hash_test.go function TestHash (line 11) | func TestHash(t *testing.T) { function TestHashFileWithSaltAndRange (line 23) | func TestHashFileWithSaltAndRange(t *testing.T) { FILE: pkg/util/mime/mime.go function init (line 95) | func init() { function ExtensionByType (line 103) | func ExtensionByType(typ string) string { function TypeByExtension (line 121) | func TypeByExtension(ext string) string { FILE: pkg/util/mime/mime_test.go function TestExtensionByType (line 9) | func TestExtensionByType(t *testing.T) { function TestTypeByExtension (line 18) | func TestTypeByExtension(t *testing.T) { FILE: pkg/util/net.go function PickFreePort (line 14) | func PickFreePort(minPort, maxPort int) (int, error) { FILE: pkg/util/overwrite_yaml.go function OverwriteYAML (line 9) | func OverwriteYAML(sourceYaml []byte, destinationYaml []byte) ([]byte, e... function traverseAndCompare (line 30) | func traverseAndCompare(sourceNode, destinationNode *yaml.Node, path str... function mapNodeToMap (line 105) | func mapNodeToMap(node *yaml.Node) map[string][]*yaml.Node { function getAllKeys (line 115) | func getAllKeys(map1, map2 map[string][]*yaml.Node) []string { function nodeKindToString (line 131) | func nodeKindToString(kind yaml.Kind) string { FILE: pkg/util/overwrite_yaml_test.go function TestOverwriteYAMLWithComments (line 39) | func TestOverwriteYAMLWithComments(t *testing.T) { function TestOverwriteYAMLWithLineComments (line 81) | func TestOverwriteYAMLWithLineComments(t *testing.T) { function TestStep1XYaml (line 124) | func TestStep1XYaml(t *testing.T) { FILE: pkg/util/platform.go function IsAppleSiliconMac (line 4) | func IsAppleSiliconMac(goos string, goarch string) bool { FILE: pkg/util/ringbuffer.go type RingBufferWriter (line 10) | type RingBufferWriter struct method Write (line 29) | func (w *RingBufferWriter) Write(p []byte) (n int, err error) { method String (line 49) | func (w *RingBufferWriter) String() string { function NewRingBufferWriter (line 20) | func NewRingBufferWriter(w io.Writer, size int) *RingBufferWriter { FILE: pkg/util/shell/net.go function WaitForPort (line 13) | func WaitForPort(port int, timeout time.Duration) error { function WaitForHTTPOK (line 29) | func WaitForHTTPOK(url string, timeout time.Duration) error { function PortIsOpen (line 51) | func PortIsOpen(port int) bool { FILE: pkg/util/shell/pipes.go type PipeFunc (line 8) | type PipeFunc type LogFunc (line 9) | type LogFunc function PipeTo (line 11) | func PipeTo(pf PipeFunc, lf LogFunc) (done chan struct{}, err error) { FILE: pkg/util/version/version.go type Version (line 9) | type Version struct method Greater (line 61) | func (v *Version) Greater(other *Version) bool { method Equal (line 76) | func (v *Version) Equal(other *Version) bool { method GreaterOrEqual (line 83) | func (v *Version) GreaterOrEqual(other *Version) bool { method EqualMinor (line 87) | func (v *Version) EqualMinor(other *Version) bool { method HasPatch (line 91) | func (v *Version) HasPatch() bool { method PatchVersion (line 95) | func (v *Version) PatchVersion() int { method Matches (line 126) | func (v *Version) Matches(other *Version) bool { function NewVersion (line 16) | func NewVersion(s string) (version *Version, err error) { function MustVersion (line 53) | func MustVersion(s string) *Version { function Equal (line 102) | func Equal(v1 string, v2 string) bool { function EqualMinor (line 106) | func EqualMinor(v1 string, v2 string) bool { function Greater (line 110) | func Greater(v1 string, v2 string) bool { function GreaterOrEqual (line 114) | func GreaterOrEqual(v1 string, v2 string) bool { function Matches (line 139) | func Matches(v1 string, v2 string) bool { function StripPatch (line 143) | func StripPatch(v string) string { function StripModifier (line 148) | func StripModifier(v string) string { FILE: pkg/util/version/version_test.go function TestVersionEqual (line 9) | func TestVersionEqual(t *testing.T) { function TestVersionGreater (line 37) | func TestVersionGreater(t *testing.T) { function TestVersionStripModifier (line 64) | func TestVersionStripModifier(t *testing.T) { function TestVersionMatches (line 71) | func TestVersionMatches(t *testing.T) { function TestVersionMatchesModifier (line 77) | func TestVersionMatchesModifier(t *testing.T) { function TestGreaterThanOrEqualToWithInvalidPatch (line 83) | func TestGreaterThanOrEqualToWithInvalidPatch(t *testing.T) { FILE: pkg/web/client.go constant pushStartURLPath (line 31) | pushStartURLPath = "/api/models/push-start" constant startChallengeURLPath (line 32) | startChallengeURLPath = "/api/models/file-challenge" type Client (line 42) | type Client struct method PostPushStart (line 138) | func (c *Client) PostPushStart(ctx context.Context, pushID string, bui... method PostNewVersion (line 171) | func (c *Client) PostNewVersion(ctx context.Context, image string, wei... method FetchAPIToken (line 221) | func (c *Client) FetchAPIToken(ctx context.Context, entity string) (st... method versionFromManifest (line 247) | func (c *Client) versionFromManifest(ctx context.Context, image string... method InitiateAndDoFileChallenge (line 368) | func (c *Client) InitiateAndDoFileChallenge(ctx context.Context, weigh... method doSingleFileChallenge (line 405) | func (c *Client) doSingleFileChallenge(ctx context.Context, file File,... type File (line 47) | type File struct type Env (line 53) | type Env struct type RuntimeConfig (line 66) | type RuntimeConfig struct type Version (line 72) | type Version struct type FileChallengeRequest (line 83) | type FileChallengeRequest struct type FileChallenge (line 88) | type FileChallenge struct type FileChallengeAnswer (line 96) | type FileChallengeAnswer struct type VersionError (line 102) | type VersionError struct type VersionErrors (line 107) | type VersionErrors struct type VersionCreate (line 114) | type VersionCreate struct type CogKey (line 118) | type CogKey struct type Keys (line 123) | type Keys struct type TokenData (line 127) | type TokenData struct function NewClient (line 131) | func NewClient(dockerCommand command.Command, client *http.Client) *Clie... function newVersionURL (line 451) | func newVersionURL(image string) (url.URL, error) { function tokenURL (line 461) | func tokenURL(entity string) url.URL { function webBaseURL (line 467) | func webBaseURL() url.URL { function codeFileName (line 474) | func codeFileName(cogConfig *config.Config, isPredict bool) (string, err... function readCode (line 495) | func readCode(cogConfig *config.Config, isPredict bool) (string, string,... function stripCodeFromStub (line 512) | func stripCodeFromStub(cogConfig *config.Config, isPredict bool) (string... function readCogConfig (line 530) | func readCogConfig(manifest *image.InspectResponse) (*config.Config, err... FILE: pkg/web/client_test.go function TestPostNewVersion (line 20) | func TestPostNewVersion(t *testing.T) { function TestVersionFromManifest (line 50) | func TestVersionFromManifest(t *testing.T) { function TestVersionURLErrorWithoutR8IMPrefix (line 79) | func TestVersionURLErrorWithoutR8IMPrefix(t *testing.T) { function TestVersionURLErrorWithout3Components (line 84) | func TestVersionURLErrorWithout3Components(t *testing.T) { function TestDoFileChallenge (line 89) | func TestDoFileChallenge(t *testing.T) { function TestFetchToken (line 171) | func TestFetchToken(t *testing.T) { FILE: pkg/weights/manifest.go type Manifest (line 15) | type Manifest struct method Save (line 50) | func (m *Manifest) Save(filename string) error { method Equal (line 65) | func (m *Manifest) Equal(other *Manifest) bool { method AddFile (line 80) | func (m *Manifest) AddFile(path string) error { type Metadata (line 20) | type Metadata struct function NewManifest (line 26) | func NewManifest() *Manifest { function LoadManifest (line 31) | func LoadManifest(filename string) (*Manifest, error) { FILE: pkg/weights/weights.go type FileWalker (line 22) | type FileWalker function FindWeights (line 24) | func FindWeights(fw FileWalker) ([]string, []string, error) { function isNonModelFiles (line 66) | func isNonModelFiles(path string) bool { constant sizeThreshold (line 80) | sizeThreshold = 10 * 1024 * 1024 function sortFilesByLevels (line 82) | func sortFilesByLevels(files []string) { function isCodeFile (line 99) | func isCodeFile(path string) bool { function isGitFile (line 104) | func isGitFile(path string) bool { function filterDirsContainingCode (line 112) | func filterDirsContainingCode(dirs []string, codeFiles []string) []string { function getDirsAndRootfiles (line 132) | func getDirsAndRootfiles(files []string) ([]string, []string) { function hasParent (line 154) | func hasParent(dir string, dirs []string) bool { FILE: pkg/weights/weights_test.go type mockFileInfo (line 13) | type mockFileInfo struct method Size (line 17) | func (mfi mockFileInfo) Size() int64 { method Name (line 20) | func (mfi mockFileInfo) Name() string { method Mode (line 23) | func (mfi mockFileInfo) Mode() os.FileMode { method ModTime (line 26) | func (mfi mockFileInfo) ModTime() time.Time { method IsDir (line 29) | func (mfi mockFileInfo) IsDir() bool { method Sys (line 32) | func (mfi mockFileInfo) Sys() any { function TestRootDirModelFiles (line 37) | func TestRootDirModelFiles(t *testing.T) { function TestSubDirModelFiles (line 53) | func TestSubDirModelFiles(t *testing.T) { function TestRootAndSubDirModelFiles (line 69) | func TestRootAndSubDirModelFiles(t *testing.T) { function TestRootDirLargeModelAndCodeFiles (line 85) | func TestRootDirLargeModelAndCodeFiles(t *testing.T) { function TestSubDirLargeModelAndCodeFiles (line 101) | func TestSubDirLargeModelAndCodeFiles(t *testing.T) { function TestSubDirLargeModelDirWithCodeFiles (line 117) | func TestSubDirLargeModelDirWithCodeFiles(t *testing.T) { function TestDirSorting (line 133) | func TestDirSorting(t *testing.T) { function TestSubDirMerge (line 149) | func TestSubDirMerge(t *testing.T) { function TestIgnoreGitFiles (line 165) | func TestIgnoreGitFiles(t *testing.T) { FILE: pkg/wheels/wheels.go function IsPreRelease (line 23) | func IsPreRelease(version string) bool { constant MinimumSDKVersion (line 29) | MinimumSDKVersion = "0.16.0" function ValidateSDKVersion (line 37) | func ValidateSDKVersion(config *WheelConfig, label string) error { type WheelSource (line 57) | type WheelSource method String (line 69) | func (s WheelSource) String() string { constant WheelSourcePyPI (line 61) | WheelSourcePyPI WheelSource = iota constant WheelSourceURL (line 63) | WheelSourceURL constant WheelSourceFile (line 65) | WheelSourceFile type WheelConfig (line 83) | type WheelConfig struct method PyPIPackageURL (line 391) | func (c *WheelConfig) PyPIPackageURL(packageName string) string { constant CogSDKWheelEnvVar (line 95) | CogSDKWheelEnvVar = "COG_SDK_WHEEL" constant CogletWheelEnvVar (line 98) | CogletWheelEnvVar = "COGLET_WHEEL" function ParseWheelValue (line 111) | func ParseWheelValue(value string) *WheelConfig { function goarchToWheelPlatform (line 143) | func goarchToWheelPlatform(goarch string) string { function bestWheelMatch (line 154) | func bestWheelMatch(matches []string, platform string) string { function distFromExecutable (line 181) | func distFromExecutable() string { function findWheelInAutoDetectDist (line 200) | func findWheelInAutoDetectDist(pattern string, platform string) string { function DetectLocalSDKVersion (line 223) | func DetectLocalSDKVersion() string { function resolveWheelPath (line 243) | func resolveWheelPath(path string, pattern string, platform string, envV... function ResolveCogWheel (line 291) | func ResolveCogWheel(envValue string, version string) (*WheelConfig, err... function GetCogWheelConfig (line 320) | func GetCogWheelConfig() (*WheelConfig, error) { function ResolveCogletWheel (line 337) | func ResolveCogletWheel(envValue string, version string, platform string... function GetCogletWheelConfig (line 366) | func GetCogletWheelConfig(targetArch string) (*WheelConfig, error) { function SemverToPEP440 (line 374) | func SemverToPEP440(version string) string { FILE: pkg/wheels/wheels_test.go function TestCogSDKWheelEnvVarName (line 11) | func TestCogSDKWheelEnvVarName(t *testing.T) { function TestWheelSourceString (line 15) | func TestWheelSourceString(t *testing.T) { function TestParseWheelValue (line 33) | func TestParseWheelValue(t *testing.T) { function TestGetCogWheelConfig (line 165) | func TestGetCogWheelConfig(t *testing.T) { function TestGetCogWheelConfigErrors (line 257) | func TestGetCogWheelConfigErrors(t *testing.T) { function TestGetCogWheelConfigAutoDetect (line 267) | func TestGetCogWheelConfigAutoDetect(t *testing.T) { function TestResolveCogWheelUsesExecutableDist (line 297) | func TestResolveCogWheelUsesExecutableDist(t *testing.T) { function TestGetCogletWheelConfig (line 336) | func TestGetCogletWheelConfig(t *testing.T) { function TestPyPIPackageURL (line 406) | func TestPyPIPackageURL(t *testing.T) { function TestIsPreRelease (line 465) | func TestIsPreRelease(t *testing.T) { function TestValidateSDKVersion (line 491) | func TestValidateSDKVersion(t *testing.T) { function TestPyPIPackageURLPreRelease (line 529) | func TestPyPIPackageURLPreRelease(t *testing.T) { FILE: python/cog/__init__.py class _ExperimentalFeatureWarning (line 53) | class _ExperimentalFeatureWarning(FutureWarning): function __getattr__ (line 63) | def __getattr__(name: str) -> object: function current_scope (line 92) | def current_scope() -> object: FILE: python/cog/_adt.py function _type_name (line 20) | def _type_name(tpe: Any) -> str: function _is_union (line 28) | def _is_union(tpe: type) -> bool: class PrimitiveType (line 40) | class PrimitiveType(Enum): method _python_type (line 54) | def _python_type() -> Dict["PrimitiveType", type | Any]: method _json_type (line 68) | def _json_type() -> Dict["PrimitiveType", str]: method _adt_type (line 82) | def _adt_type() -> Dict[type | Any, "PrimitiveType"]: method from_type (line 95) | def from_type(tpe: type | Any) -> "PrimitiveType": method normalize (line 111) | def normalize(self, value: Any) -> Any: method python_type_name (line 161) | def python_type_name(self) -> str: method json_type (line 165) | def json_type(self) -> Dict[str, Any]: method json_encode (line 176) | def json_encode(self, value: Any) -> Any: class Repetition (line 192) | class Repetition(Enum): class FieldType (line 201) | class FieldType: method from_type (line 209) | def from_type(tpe: type) -> "FieldType": method normalize (line 273) | def normalize(self, value: Any) -> Any: method json_type (line 283) | def json_type(self) -> Dict[str, Any]: method json_encode (line 289) | def json_encode(self, value: Any) -> Any: method json_decode (line 299) | def json_decode(self, value: Any) -> Any: method python_type_name (line 309) | def python_type_name(self) -> str: class InputField (line 321) | class InputField: class OutputKind (line 338) | class OutputKind(Enum): class OutputType (line 349) | class OutputType: method json_type (line 357) | def json_type(self) -> Dict[str, Any]: method normalize (line 406) | def normalize(self, value: Any) -> Any: method json_encode (line 410) | def json_encode(self, value: Any) -> Any: method _transform (line 426) | def _transform(self, value: Any, json: bool) -> Any: class PredictorInfo (line 461) | class PredictorInfo: FILE: python/cog/_inspector.py function _check_parent (line 32) | def _check_parent(child: type, parent: type) -> bool: function _type_name (line 37) | def _type_name(tpe: Any) -> str: function _validate_setup (line 45) | def _validate_setup(f: Callable[..., Any]) -> None: function _validate_predict (line 75) | def _validate_predict(f: Callable[..., Any], f_name: str, is_class_fn: b... function _validate_input_constraints (line 96) | def _validate_input_constraints( function _create_input_field (line 194) | def _create_input_field( class _AnyType (line 250) | class _AnyType: method normalize (line 254) | def normalize(value: Any) -> Any: method json_type (line 258) | def json_type() -> Dict[str, Any]: method json_encode (line 262) | def json_encode(value: Any) -> Any: function _create_output_type (line 269) | def _create_output_type(tpe: type) -> adt.OutputType: function _create_predictor_info (line 332) | def _create_predictor_info( function _unwrap (line 369) | def _unwrap(f: Callable[..., Any]) -> Callable[..., Any]: function _is_coder (line 386) | def _is_coder(cls: Type[Any]) -> bool: function _find_coders (line 391) | def _find_coders(module: ModuleType) -> None: function create_predictor (line 403) | def create_predictor(module_name: str, predictor_name: str) -> adt.Predi... function check_input (line 458) | def check_input( FILE: python/cog/_schemas.py function to_json_input (line 15) | def to_json_input(predictor: adt.PredictorInfo) -> Dict[str, Any]: function to_json_enums (line 91) | def to_json_enums(predictor: adt.PredictorInfo) -> Dict[str, Any]: function to_json_output (line 110) | def to_json_output(predictor: adt.PredictorInfo) -> Dict[str, Any]: function to_json_schema (line 115) | def to_json_schema( FILE: python/cog/coder.py class Coder (line 12) | class Coder: method register (line 43) | def register(coder: Type["Coder"]) -> None: method lookup (line 53) | def lookup(tpe: type | Any) -> Optional["Coder"]: method factory (line 71) | def factory(tpe: Type[Any]) -> Optional["Coder"]: method encode (line 87) | def encode(self, x: Any) -> Dict[str, Any]: method decode (line 100) | def decode(self, x: Dict[str, Any]) -> Any: FILE: python/cog/command/openapi_schema.py function _load_module_from_ref (line 23) | def _load_module_from_ref(ref: str) -> tuple[str, str]: function remove_title_next_to_ref (line 45) | def remove_title_next_to_ref( function fix_nullable_anyof (line 63) | def fix_nullable_anyof(schema_node: Any) -> None: FILE: python/cog/config.py class Config (line 22) | class Config: method __init__ (line 25) | def __init__(self, config: Optional[dict[str, Any]] = None) -> None: method _cog_config (line 29) | def _cog_config(self) -> dict[str, Any]: method predictor_predict_ref (line 44) | def predictor_predict_ref(self) -> Optional[str]: method predictor_train_ref (line 51) | def predictor_train_ref(self) -> Optional[str]: method get_predictor_ref (line 57) | def get_predictor_ref(self, mode: Mode) -> str: FILE: python/cog/errors.py class CogError (line 1) | class CogError(Exception): class ConfigDoesNotExist (line 5) | class ConfigDoesNotExist(CogError): class PredictorNotSet (line 9) | class PredictorNotSet(CogError): FILE: python/cog/input.py class FieldInfo (line 13) | class FieldInfo: function Input (line 32) | def Input( FILE: python/cog/mode.py class Mode (line 4) | class Mode(Enum): method __str__ (line 10) | def __str__(self) -> str: FILE: python/cog/model.py class BaseModel (line 12) | class BaseModel: method __init_subclass__ (line 39) | def __init_subclass__( FILE: python/cog/predictor.py class BasePredictor (line 18) | class BasePredictor: method setup (line 38) | def setup( method predict (line 53) | def predict(self, **kwargs: Any) -> Any: method scope (line 73) | def scope(self) -> Any: method record_metric (line 88) | def record_metric(self, key: str, value: Any, mode: str = "replace") -... function load_predictor_from_ref (line 115) | def load_predictor_from_ref(ref: str) -> BasePredictor: function has_setup_weights (line 136) | def has_setup_weights(predictor: BasePredictor) -> bool: function extract_setup_weights (line 144) | def extract_setup_weights(predictor: BasePredictor) -> Optional[Union[Pa... FILE: python/cog/server/http.py class Mode (line 9) | class Mode(Enum): method __str__ (line 13) | def __str__(self) -> str: FILE: python/cog/suppress_output.py function suppress_output (line 8) | def suppress_output() -> Iterator[None]: FILE: python/cog/types.py function _len_bytes (line 38) | def _len_bytes(s: str) -> int: function _truncate_filename_bytes (line 43) | def _truncate_filename_bytes(filename: str, length: int) -> str: function get_filename (line 61) | def get_filename(url: str) -> str: class Secret (line 94) | class Secret: method __repr__ (line 109) | def __repr__(self) -> str: method __str__ (line 112) | def __str__(self) -> str: method get_secret_value (line 115) | def get_secret_value(self) -> Optional[str]: class URLFile (line 125) | class URLFile(io.IOBase): method __init__ (line 134) | def __init__(self, url: str, filename: Optional[str] = None) -> None: method __del__ (line 147) | def __del__(self) -> None: method __getstate__ (line 159) | def __getstate__(self) -> Dict[str, Any]: method __setstate__ (line 165) | def __setstate__(self, state: Dict[str, Any]) -> None: method __setattr__ (line 170) | def __setattr__(self, name: str, value: Any) -> None: method __getattr__ (line 176) | def __getattr__(self, name: str) -> Any: method __delattr__ (line 183) | def __delattr__(self, name: str) -> None: method __iter__ (line 190) | def __iter__(self) -> Iterator[bytes]: method __wrapped__ (line 194) | def __wrapped__(self) -> Any: method __repr__ (line 211) | def __repr__(self) -> str: class File (line 225) | class File(io.IOBase): method validate (line 233) | def validate(cls, value: Any) -> io.IOBase: class URLPath (line 256) | class URLPath(pathlib.PosixPath): method __init__ (line 268) | def __init__(self, *, source: str, filename: str, fileobj: io.IOBase) ... method __new__ (line 278) | def __new__(cls, *, source: str, filename: str, fileobj: io.IOBase) ->... method convert (line 284) | def convert(self) -> "Path": method unlink (line 294) | def unlink(self, missing_ok: bool = False) -> None: method __str__ (line 299) | def __str__(self) -> str: class Path (line 310) | class Path(pathlib.PosixPath): method validate (line 324) | def validate(cls, value: Any) -> pathlib.Path: class ConcatenateIterator (line 347) | class ConcatenateIterator(Iterator[Item]): method __next__ (line 361) | def __next__(self) -> Item: ... class AsyncConcatenateIterator (line 364) | class AsyncConcatenateIterator(AsyncIterator[Item]): method __anext__ (line 378) | async def __anext__(self) -> Item: ... FILE: python/tests/test_emit_metric.py class TestEmitMetric (line 7) | class TestEmitMetric: method test_import_succeeds (line 10) | def test_import_succeeds(self) -> None: method test_attribute_access_succeeds (line 19) | def test_attribute_access_succeeds(self) -> None: method test_prints_deprecation_to_stderr (line 28) | def test_prints_deprecation_to_stderr(self) -> None: method test_message_prints_once (line 37) | def test_message_prints_once(self) -> None: method test_callable (line 47) | def test_callable(self) -> None: method test_module_attribute_callable (line 60) | def test_module_attribute_callable(self) -> None: method test_unknown_attr_still_raises (line 75) | def test_unknown_attr_still_raises(self) -> None: FILE: python/tests/test_experimental_feature_warning.py class TestExperimentalFeatureWarning (line 7) | class TestExperimentalFeatureWarning: method test_import_succeeds (line 10) | def test_import_succeeds(self) -> None: method test_prints_deprecation_to_stderr (line 19) | def test_prints_deprecation_to_stderr(self) -> None: method test_message_prints_once (line 28) | def test_message_prints_once(self) -> None: method test_is_future_warning_subclass (line 41) | def test_is_future_warning_subclass(self) -> None: method test_filterwarnings_compat (line 54) | def test_filterwarnings_compat(self) -> None: method test_unknown_attr_raises (line 68) | def test_unknown_attr_raises(self) -> None: FILE: python/tests/test_input.py class TestInput (line 9) | class TestInput: method test_input_returns_fieldinfo (line 12) | def test_input_returns_fieldinfo(self) -> None: method test_input_with_default (line 16) | def test_input_with_default(self) -> None: method test_input_with_numeric_constraints (line 21) | def test_input_with_numeric_constraints(self) -> None: method test_input_with_string_constraints (line 27) | def test_input_with_string_constraints(self) -> None: method test_input_with_choices (line 33) | def test_input_with_choices(self) -> None: method test_input_with_deprecated (line 38) | def test_input_with_deprecated(self) -> None: method test_input_default_factory_raises_error (line 42) | def test_input_default_factory_raises_error(self) -> None: method test_input_immutable_defaults_stored_directly (line 46) | def test_input_immutable_defaults_stored_directly(self) -> None: method test_input_no_default (line 51) | def test_input_no_default(self) -> None: class TestFieldInfo (line 58) | class TestFieldInfo: method test_fieldinfo_is_frozen (line 61) | def test_fieldinfo_is_frozen(self) -> None: method test_fieldinfo_defaults (line 66) | def test_fieldinfo_defaults(self) -> None: method test_fieldinfo_none_defaults (line 73) | def test_fieldinfo_none_defaults(self) -> None: FILE: python/tests/test_model.py class TestBaseModel (line 9) | class TestBaseModel: method test_subclass_becomes_dataclass (line 12) | def test_subclass_becomes_dataclass(self) -> None: method test_subclass_can_be_instantiated (line 19) | def test_subclass_can_be_instantiated(self) -> None: method test_subclass_with_defaults (line 28) | def test_subclass_with_defaults(self) -> None: method test_subclass_with_optional (line 37) | def test_subclass_with_optional(self) -> None: method test_nested_models (line 46) | def test_nested_models(self) -> None: method test_inheritance (line 59) | def test_inheritance(self) -> None: method test_auto_dataclass_false (line 70) | def test_auto_dataclass_false(self) -> None: method test_primary_base_must_be_basemodel (line 84) | def test_primary_base_must_be_basemodel(self) -> None: method test_cannot_mixin_dataclass (line 97) | def test_cannot_mixin_dataclass(self) -> None: method test_auto_dataclass_inheritance_mismatch (line 113) | def test_auto_dataclass_inheritance_mismatch(self) -> None: method test_basemodel_asdict (line 127) | def test_basemodel_asdict(self) -> None: FILE: python/tests/test_predictor.py class TestBasePredictor (line 8) | class TestBasePredictor: method test_subclass_can_override_predict (line 11) | def test_subclass_can_override_predict(self) -> None: method test_default_predict_raises (line 20) | def test_default_predict_raises(self) -> None: method test_setup_is_optional (line 28) | def test_setup_is_optional(self) -> None: method test_setup_with_weights (line 38) | def test_setup_with_weights(self) -> None: method test_setup_with_path_weights (line 52) | def test_setup_with_path_weights(self) -> None: method test_predictor_with_multiple_inputs (line 66) | def test_predictor_with_multiple_inputs(self) -> None: method test_predictor_with_state (line 78) | def test_predictor_with_state(self) -> None: FILE: python/tests/test_types.py class TestSecret (line 16) | class TestSecret: method test_secret_creation (line 19) | def test_secret_creation(self) -> None: method test_secret_masks_in_str (line 23) | def test_secret_masks_in_str(self) -> None: method test_secret_masks_in_repr (line 28) | def test_secret_masks_in_repr(self) -> None: method test_secret_none_value (line 33) | def test_secret_none_value(self) -> None: method test_secret_default_none (line 38) | def test_secret_default_none(self) -> None: method test_secret_is_dataclass (line 42) | def test_secret_is_dataclass(self) -> None: method test_secret_is_frozen (line 45) | def test_secret_is_frozen(self) -> None: class TestPath (line 54) | class TestPath: method test_path_from_string (line 57) | def test_path_from_string(self) -> None: method test_path_is_pathlib_subclass (line 61) | def test_path_is_pathlib_subclass(self) -> None: class TestFile (line 68) | class TestFile: method test_file_validate_iobase (line 71) | def test_file_validate_iobase(self) -> None: method test_file_validate_data_uri (line 76) | def test_file_validate_data_uri(self) -> None: method test_file_validate_invalid_scheme (line 83) | def test_file_validate_invalid_scheme(self) -> None: class TestURLFile (line 91) | class TestURLFile: method test_urlfile_creation (line 94) | def test_urlfile_creation(self) -> None: method test_urlfile_invalid_scheme (line 99) | def test_urlfile_invalid_scheme(self) -> None: method test_urlfile_picklable (line 106) | def test_urlfile_picklable(self) -> None: method test_urlfile_custom_filename (line 115) | def test_urlfile_custom_filename(self) -> None: class TestIterators (line 121) | class TestIterators: method test_concatenate_iterator_is_abstract (line 124) | def test_concatenate_iterator_is_abstract(self) -> None: method test_async_concatenate_iterator_is_abstract (line 130) | def test_async_concatenate_iterator_is_abstract(self) -> None: FILE: test-helpers/https-server/main.go function main (line 15) | func main() { FILE: test-integration/test_integration/fixtures/hello-image/predict.py class Predictor (line 4) | class Predictor(BasePredictor): method predict (line 5) | def predict(self, word: str) -> Path: FILE: tools/compatgen/internal/cuda.go function FetchCUDABaseImages (line 20) | func FetchCUDABaseImages(ctx context.Context) ([]config.CUDABaseImage, e... function fetchCUDABaseImageTags (line 49) | func fetchCUDABaseImageTags(url string) ([]string, error) { function parseCUDABaseImage (line 94) | func parseCUDABaseImage(ctx context.Context, tag string) (*config.CUDABa... FILE: tools/compatgen/internal/tensorflow.go function FetchTensorFlowCompatibilityMatrix (line 15) | func FetchTensorFlowCompatibilityMatrix() ([]config.TFCompatibility, err... function parsePythonVersionsCell (line 63) | func parsePythonVersionsCell(val string) ([]string, error) { function newVersion (line 92) | func newVersion(major int, minor int) string { function splitPythonVersion (line 96) | func splitPythonVersion(version string) (major int, minor int, err error) { FILE: tools/compatgen/internal/torch.go function FetchTorchCompatibilityMatrix (line 24) | func FetchTorchCompatibilityMatrix() ([]config.TorchCompatibility, error) { function FetchTorchPackages (line 55) | func FetchTorchPackages(name string) ([]TorchPackage, error) { function getLatestVersion (line 60) | func getLatestVersion(packages []TorchPackage) string { function fetchCurrentTorchVersions (line 75) | func fetchCurrentTorchVersions(compats []config.TorchCompatibility) ([]c... function parseTorchInstallString (line 129) | func parseTorchInstallString(s string, defaultVersions map[string]string... function fetchPreviousTorchVersions (line 200) | func fetchPreviousTorchVersions(compats []config.TorchCompatibility) ([]... function parsePreviousTorchVersionsCode (line 225) | func parsePreviousTorchVersionsCode(code string, compats []config.TorchC... function fixTorchCompatibility (line 278) | func fixTorchCompatibility(compat *config.TorchCompatibility) { function basePytorchURL (line 284) | func basePytorchURL() string { function pytorchURL (line 288) | func pytorchURL(name string) string { function ExtractSubFeaturesFromPytorchVersion (line 293) | func ExtractSubFeaturesFromPytorchVersion(pytorchVersion string) (string... function FindCompatiblePythonVersions (line 335) | func FindCompatiblePythonVersions(torchVersion string, torchVisionVersio... function fetchTorchPackagesFromURL (line 400) | func fetchTorchPackagesFromURL(url string) ([]TorchPackage, error) { function findTorchPackagesWithVersion (line 470) | func findTorchPackagesWithVersion(pkgName string, url string, version st... FILE: tools/compatgen/internal/torch_package.go type TorchPackage (line 3) | type TorchPackage struct method Equals (line 11) | func (c *TorchPackage) Equals(other TorchPackage) bool { FILE: tools/compatgen/internal/torch_test.go function TestFetchTorchPackages (line 17) | func TestFetchTorchPackages(t *testing.T) { function TestIsValidPytorchVersionFormat (line 74) | func TestIsValidPytorchVersionFormat(t *testing.T) { function TestIsValidPytorchVersionFormatWithOldVersion (line 84) | func TestIsValidPytorchVersionFormatWithOldVersion(t *testing.T) { function TestIsValidPytorchAudioVersionFormat (line 94) | func TestIsValidPytorchAudioVersionFormat(t *testing.T) { function TestIsValidPytorchAudioVersionFormatBasic (line 104) | func TestIsValidPytorchAudioVersionFormatBasic(t *testing.T) { function TestIsValidPytorchVisionVersionFormatPostRelease (line 114) | func TestIsValidPytorchVisionVersionFormatPostRelease(t *testing.T) { function TestIsValidPytorchVisionEarlyVersion (line 124) | func TestIsValidPytorchVisionEarlyVersion(t *testing.T) { function TestIsValidPytorchAudioEarlyVersion (line 134) | func TestIsValidPytorchAudioEarlyVersion(t *testing.T) { function TestURLEncodedVersion (line 144) | func TestURLEncodedVersion(t *testing.T) { function TestVersionUnderFolder (line 154) | func TestVersionUnderFolder(t *testing.T) { function TestPythonMVersion (line 164) | func TestPythonMVersion(t *testing.T) { FILE: tools/compatgen/internal/util.go function split2 (line 7) | func split2(s string, sep string) (string, string) { FILE: tools/compatgen/main.go function main (line 14) | func main() { FILE: tools/gendocs/main.go function main (line 18) | func main() { function generateDocs (line 38) | func generateDocs(outputPath string) error { function processCommandDoc (line 108) | func processCommandDoc(content string, fileName string) string { function removeConsecutiveBlankLines (line 196) | func removeConsecutiveBlankLines(lines []string) []string { function hasSynopsisSection (line 212) | func hasSynopsisSection(lines []string) bool { FILE: tools/test-harness/harness/cli.py function main (line 17) | def main(argv: list[str] | None = None) -> None: function _cmd_list (line 75) | def _cmd_list(args: argparse.Namespace) -> None: function _cmd_build (line 88) | def _cmd_build(args: argparse.Namespace) -> None: function _cmd_run (line 141) | def _cmd_run(args: argparse.Namespace) -> None: function _add_common_args (line 212) | def _add_common_args(parser: argparse.ArgumentParser) -> None: function _load_manifest (line 267) | def _load_manifest(manifest_path: str | None) -> dict: function _filter_models (line 282) | def _filter_models(manifest: dict, args: argparse.Namespace) -> list[dict]: FILE: tools/test-harness/harness/cog_resolver.py function resolve_latest_stable_version (line 27) | def resolve_latest_stable_version() -> str: function _platform_asset_name (line 59) | def _platform_asset_name() -> str: function download_cog_binary (line 81) | def download_cog_binary(tag: str, dest_dir: Path | None = None) -> Path: function resolve_cog_binary (line 118) | def resolve_cog_binary( function resolve_latest_sdk_version (line 167) | def resolve_latest_sdk_version() -> str: function resolve_sdk_version (line 181) | def resolve_sdk_version( FILE: tools/test-harness/harness/patcher.py function deep_merge (line 12) | def deep_merge(base: dict[str, Any], override: dict[str, Any]) -> dict[s... function patch_cog_yaml (line 23) | def patch_cog_yaml( FILE: tools/test-harness/harness/report.py function console_report (line 13) | def console_report( function json_report (line 78) | def json_report( function write_json_report (line 139) | def write_json_report( function _write (line 155) | def _write( function _timing_str (line 163) | def _timing_str(build_s: float, tests: list[Any]) -> str: FILE: tools/test-harness/harness/runner.py class TestCaseResult (line 25) | class TestCaseResult: class ModelResult (line 33) | class ModelResult: class Runner (line 48) | class Runner: method __init__ (line 51) | def __init__( method prepare_model (line 69) | def prepare_model(self, model: dict[str, Any]) -> Path: method build_model (line 73) | def build_model(self, model_dir: Path, model: dict[str, Any]) -> None: method run_model (line 77) | def run_model(self, model: dict[str, Any]) -> ModelResult: method _prepare_model (line 133) | def _prepare_model(self, model: dict[str, Any]) -> Path: method _clone_repo (line 155) | def _clone_repo(self, repo: str) -> Path: method _cog_build (line 175) | def _cog_build(self, model_dir: Path, model: dict[str, Any]) -> None: method _run_predict_test (line 193) | def _run_predict_test( method _run_train_test (line 259) | def _run_train_test( method _extract_output (line 326) | def _extract_output(proc: subprocess.CompletedProcess[str], model_dir:... method _resolve_input (line 351) | def _resolve_input(self, value: Any) -> str: method _build_env (line 368) | def _build_env(self, model: dict[str, Any]) -> dict[str, str]: method cleanup (line 376) | def cleanup(self) -> None: FILE: tools/test-harness/harness/validators.py class ValidationResult (line 14) | class ValidationResult: function validate (line 19) | def validate(output: str, expect: dict[str, Any]) -> ValidationResult: function _validate_exact (line 34) | def _validate_exact(output: str, expect: dict[str, Any]) -> ValidationRe... function _validate_contains (line 45) | def _validate_contains(output: str, expect: dict[str, Any]) -> Validatio... function _validate_regex (line 55) | def _validate_regex(output: str, expect: dict[str, Any]) -> ValidationRe... function _validate_file_exists (line 65) | def _validate_file_exists(output: str, expect: dict[str, Any]) -> Valida... function _validate_json_match (line 98) | def _validate_json_match(output: str, expect: dict[str, Any]) -> Validat... function _validate_json_keys (line 117) | def _validate_json_keys(output: str, expect: dict[str, Any]) -> Validati... function _validate_not_empty (line 153) | def _validate_not_empty(output: str, _expect: dict[str, Any]) -> Validat... function _is_subset (line 162) | def _is_subset(subset: Any, superset: Any) -> bool: FILE: tools/test-registry-util/main.go function main (line 48) | func main() { function runAndInit (line 102) | func runAndInit(ctx context.Context, dstDir string) error { function runAndCatalog (line 218) | func runAndCatalog(ctx context.Context, dir string) error { function catalog (line 241) | func catalog(ctx context.Context, addr string) error { function startRegistryTC (line 303) | func startRegistryTC(ctx context.Context, dir string) (*registry.Registr... FILE: tools/weights-gen/main.go function main (line 20) | func main() { function parseSize (line 94) | func parseSize(s string) (int64, error) { function generateWeights (line 131) | func generateWeights(outputDir, destPrefix, outputPath string, count int... function generateRandomFile (line 211) | func generateRandomFile(path string, size int64, rng *rand.Rand) error { function formatSize (line 239) | func formatSize(bytes int64) string { function processFile (line 258) | func processFile(path, baseDir, destPrefix string) (*model.WeightFile, e...