Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/amazon-science/chronos-forecasting
/ functions
Functions
323 in github.com/amazon-science/chronos-forecasting
⨍
Functions
323
◇
Types & classes
43
↳
Endpoints
26
Function
test_attention_implementations_with_output_attentions
Test that attention implementations handle output_attentions correctly.
test/test_chronos2.py:1104
Function
test_base_chronos2_pipeline_loads_from_hf
()
test/test_chronos2.py:39
Function
test_base_chronos2_pipeline_loads_from_s3
()
test/test_chronos2.py:35
Function
test_base_chronos_pipeline_loads_from_huggingface
()
test/test_chronos_bolt.py:25
Function
test_base_chronos_pipeline_loads_from_huggingface
()
test/test_chronos.py:27
Function
test_chronos2_lora_pipeline_loads_from_disk
()
test/test_chronos2.py:43
Function
test_convert_df_input_to_list_of_dicts_input_no_warning_when_validation_disabled
The AutoGluon path uses validate_inputs=False and must not emit the deprecation warning.
test/test_df_utils.py:353
Function
test_convert_df_input_to_list_of_dicts_input_produces_expected_output
Lock down the legacy output contract: structure, item ordering, value preservation, and prediction timestamps (multi-target, with past-only + know
test/test_df_utils.py:294
Function
test_eager_and_sdpa_produce_identical_outputs
Test that eager and SDPA implementations produce identical outputs on full pipeline.
test/test_chronos2.py:1147
Function
test_from_data_frame_accepts_short_series
Series with fewer than 3 points are no longer rejected (freq inference happens elsewhere).
test/test_preprocess.py:313
Function
test_from_data_frame_does_not_mutate_caller_dataframes
()
test/test_preprocess.py:303
Function
test_from_data_frame_raises_for_non_numeric_target
()
test/test_preprocess.py:321
Function
test_from_data_frame_raises_for_unknown_known_covariates_names
()
test/test_preprocess.py:292
Function
test_from_data_frame_raises_when_future_df_and_known_covariates_names_both_given
()
test/test_preprocess.py:273
Function
test_from_data_frame_treats_bool_covariate_as_categorical
Bool covariates must be target-encoded as categories, not passed through as raw 0/1 floats.
test/test_preprocess.py:349
Function
test_from_data_frame_with_categorical_covariate_target_encoding
Target encoding with single target should produce finite, per-item-aware encodings.
test/test_preprocess.py:328
Function
test_from_data_frame_with_past_and_future_covariates
()
test/test_preprocess.py:251
Function
test_from_data_frame_with_single_target
()
test/test_preprocess.py:238
Function
test_from_list_of_dicts_categorical_unseen_future_does_not_leak_category_zero
End-to-end: unseen future category should encode to item mean, not category 0's encoding.
test/test_preprocess.py:474
Function
test_from_list_of_dicts_does_not_mutate_caller_data
()
test/test_preprocess.py:165
Function
test_from_list_of_dicts_nan_is_its_own_target_encoded_category
NaN is a category with its own mean: distinct from "x", from the item mean, and matching itself across past and future.
test/test_preprocess.py:522
Function
test_from_list_of_dicts_orders_past_only_before_known_future
context[-n_future_covariates:] must contain known-future rows (Chronos2Dataset invariant).
test/test_preprocess.py:102
Function
test_from_list_of_dicts_ordinal_encoding_unseen_future_is_nan
()
test/test_preprocess.py:561
Function
test_from_list_of_dicts_ordinal_encoding_when_multivariate
With multi-target, falls back to ordinal encoding (not target encoding).
test/test_preprocess.py:501
Function
test_from_list_of_dicts_preserves_numeric_covariate_values
()
test/test_preprocess.py:215
Function
test_from_list_of_dicts_raises_for_unknown_known_covariates_names
()
test/test_preprocess.py:182
Function
test_from_list_of_dicts_raises_when_future_keys_not_subset_of_past
()
test/test_preprocess.py:193
Function
test_from_list_of_dicts_raises_when_target_too_many_dims
()
test/test_preprocess.py:205
Function
test_from_list_of_dicts_returns_empty_for_empty_input
()
test/test_preprocess.py:211
Function
test_from_list_of_dicts_translates_none_future_to_known_covariates
{'a': None} in future_covariates should behave like known_covariates_names=['a'].
test/test_preprocess.py:149
Function
test_from_list_of_dicts_unseen_future_string_falls_back_to_item_mean
An unseen future string encodes to the item mean, not the NaN category.
test/test_preprocess.py:542
Function
test_from_list_of_dicts_with_known_covariates_names_nan_fills_future
()
test/test_preprocess.py:129
Function
test_from_list_of_dicts_with_target_only
()
test/test_preprocess.py:87
Function
test_from_list_of_tensors_handles_mixed_1d_and_2d
()
test/test_preprocess.py:59
Function
test_from_list_of_tensors_raises_for_3d_element
()
test/test_preprocess.py:78
Function
test_from_tensor_accepts_numpy_array
()
test/test_preprocess.py:41
Function
test_from_tensor_raises_for_wrong_ndim
()
test/test_preprocess.py:50
Function
test_from_tensor_returns_one_prepared_input_per_series
(dtype)
test/test_preprocess.py:23
Function
test_given_even_data_and_strides_patch_operator_output_is_correct
()
test/test_chronos_bolt.py:271
Function
test_given_even_data_patch_operator_output_is_correct
()
test/test_chronos_bolt.py:247
Function
test_given_uneven_data_patch_operator_pads_and_output_is_correct
()
test/test_chronos_bolt.py:289
Function
test_infer_freq_from_df_ignores_series_with_fewer_than_three_points
()
test/test_df_utils.py:30
Function
test_infer_freq_from_df_raises_for_irregular_series
()
test/test_df_utils.py:42
Function
test_infer_freq_from_df_raises_when_no_series_has_three_points
()
test/test_df_utils.py:36
Function
test_infer_freq_from_df_raises_when_series_disagree
()
test/test_df_utils.py:54
Function
test_infer_freq_from_df_recovers_the_data_frequency
(freq)
test/test_df_utils.py:24
Function
test_interpolate_quantiles
(query_quantiles, orig_quantiles, orig_values, expected_values)
test/test_utils.py:65
Function
test_make_future_df_generates_correct_timestamps
(freq)
test/test_df_utils.py:71
Function
test_make_future_df_infers_freq_when_not_provided
()
test/test_df_utils.py:96
Function
test_make_future_df_preserves_first_appearance_item_order
()
test/test_df_utils.py:89
Function
test_normalize_df_coerces_string_timestamps_to_datetime
()
test/test_df_utils.py:125
Function
test_normalize_df_does_not_mutate_caller_dataframe
()
test/test_df_utils.py:155
Function
test_normalize_df_groups_by_id_and_sorts_by_timestamp
()
test/test_df_utils.py:107
Function
test_normalize_df_raises_for_ids_absent_from_order
()
test/test_df_utils.py:143
Function
test_normalize_df_respects_explicit_order
()
test/test_df_utils.py:131
Function
test_pad_and_stack
(tensors: list)
test/test_utils.py:21
Function
test_pipeline_can_be_finetuned_with_empty_future_covariates
( pipeline, inputs, prediction_length, expected_output_shapes, ft_future_values )
test/test_chronos2.py:907
Function
test_pipeline_can_be_finetuned_with_preprocessed_hf_dataset
Test that fine-tuning works with preprocessed inputs from a HuggingFace Dataset.
test/test_chronos2.py:1202
Function
test_pipeline_can_be_finetuned_with_validation
(pipeline, inputs, prediction_length, expected_output_shapes)
test/test_chronos2.py:850
Function
test_pipeline_can_evaluate_on_dummy_fev_task
(task_kwargs)
test/test_chronos_bolt.py:359
Function
test_pipeline_can_evaluate_on_dummy_fev_task
(pipeline, task_kwargs)
test/test_chronos2.py:392
Function
test_pipeline_embed
(model_dtype: torch.dtype, input_dtype: torch.dtype)
test/test_chronos_bolt.py:130
Function
test_pipeline_embed
(model_dtype: torch.dtype, input_dtype: torch.dtype)
test/test_chronos.py:287
Function
test_pipeline_predict
(torch_dtype: torch.dtype, input_dtype: torch.dtype)
test/test_chronos_bolt.py:31
Function
test_pipeline_predict
(model_dtype: torch.dtype, input_dtype: torch.dtype)
test/test_chronos.py:178
Function
test_pipeline_predict_can_handle_different_model_and_input_dtypes
(dtype: torch.dtype, input_dtype: torch.dtype)
test/test_chronos2.py:314
Function
test_pipeline_predict_quantiles
( torch_dtype: torch.dtype, input_dtype: torch.dtype, prediction_length: int, quantile_levels:
test/test_chronos_bolt.py:86
Function
test_pipeline_predict_quantiles
( model_dtype: torch.dtype, input_dtype: torch.dtype, prediction_length: int, quantile_levels:
test/test_chronos.py:240
Function
test_pipeline_works_with_different_attention_implementations
Test that the pipeline works with different attention implementations.
test/test_chronos2.py:1077
Function
test_predict_df_accepts_short_series_with_explicit_freq
Series with fewer than 3 points are allowed as long as the frequency can be determined.
test/test_chronos2.py:483
Function
test_predict_df_df_validation_errors
(pipeline, context_data, error_match)
test/test_chronos2.py:476
Function
test_predict_df_future_df_validation_errors
(pipeline, future_data, error_match)
test/test_chronos2.py:516
Function
test_predict_df_matches_manual_preprocess_path
predict_df must produce the same output as manually preprocessing the df with from_data_frame + make_future_df and assembling the result (multi-ta
test/test_chronos2.py:598
Function
test_predict_df_outputs_different_results_with_cross_learning_enabled
( pipeline, context_setup, future_setup, prediction_length )
test/test_chronos2.py:688
Function
test_predict_df_with_future_df_missing_series_raises_error
(pipeline)
test/test_chronos2.py:568
Function
test_predict_df_with_future_df_with_different_freq_raises_error
(pipeline)
test/test_chronos2.py:578
Function
test_predict_df_with_future_df_with_different_lengths_raises_error
(pipeline)
test/test_chronos2.py:588
Function
test_predict_df_with_inconsistent_frequencies_raises_error
(pipeline)
test/test_chronos_bolt.py:218
Function
test_predict_df_with_inconsistent_frequencies_raises_error
(pipeline)
test/test_chronos.py:370
Function
test_predict_df_with_inconsistent_frequencies_raises_error
(pipeline)
test/test_chronos2.py:544
Function
test_predict_df_with_non_uniform_timestamps_raises_error
(pipeline)
test/test_chronos_bolt.py:198
Function
test_predict_df_with_non_uniform_timestamps_raises_error
(pipeline)
test/test_chronos.py:350
Function
test_predict_df_with_non_uniform_timestamps_raises_error
(pipeline, validate_inputs)
test/test_chronos2.py:524
Function
test_predict_df_works_for_valid_inputs
(pipeline, context_setup, expected_rows, freq)
test/test_chronos_bolt.py:173
Function
test_predict_df_works_for_valid_inputs
(pipeline, context_setup, expected_rows, freq)
test/test_chronos.py:325
Function
test_predict_df_works_for_valid_inputs
( pipeline, context_setup, future_setup, freq, validate_inputs, prediction_length )
test/test_chronos2.py:428
Function
test_target_encode_handles_nans_in_target
()
test/test_preprocess.py:428
Function
test_target_encode_item_with_all_nan_targets_does_not_crash
()
test/test_preprocess.py:457
Function
test_target_encode_returns_none_future_when_not_provided
()
test/test_preprocess.py:444
Function
test_target_encode_seen_category_uses_smoothed_mean
()
test/test_preprocess.py:405
Function
test_target_encode_unseen_future_category_falls_back_to_item_mean
Unseen categories at inference must map to per-item mean, not category 0's encoding.
test/test_preprocess.py:377
Function
test_token_clipping
(n_tokens)
test/test_chronos.py:424
Function
test_tokenizer_consistency
(n_numerical_tokens: int, n_special_tokens: int)
test/test_chronos.py:33
Function
test_tokenizer_fixed_data
(n_numerical_tokens: int, n_special_tokens: int, use_eos_token: bool)
test/test_chronos.py:73
Function
test_tokenizer_number_of_buckets
(n_tokens)
test/test_chronos.py:395
Function
test_tokenizer_random_data
(use_eos_token: bool)
test/test_chronos.py:124
Function
test_train_dataset_does_not_materialize_lazy_inputs
()
test/test_chronos2.py:1248
Function
test_train_dataset_raises_when_all_lazy_inputs_too_short
()
test/test_chronos2.py:1268
Function
test_two_step_finetuning_with_df_input_works
(pipeline, context_setup, future_setup, freq)
test/test_chronos2.py:1002
Function
test_validate_and_normalize_df_aligns_future_df_order_to_df
()
test/test_df_utils.py:266
← previous
next →
201–300 of 323, ranked by callers