• xeron56's avatar
    Add CUDA device compatibility checks to TimesFM backend · 6132f5c0
    xeron56 authored
    - Implemented a function to check if the current CUDA device is supported by the installed PyTorch version.
    - Added warnings for unsupported CUDA devices, informing users that the CPU will be used instead.
    - Updated the TimesFMBackend class to utilize the new compatibility checks and handle GPU names appropriately.
    - Introduced unit tests to validate the behavior of the CUDA compatibility function with various simulated CUDA environments.
    6132f5c0
Name
Last commit
Last update
..
__init__.py Loading commit data...
conftest.py Loading commit data...
test_alpha_vantage_hardening.py Loading commit data...
test_analysis_api.py Loading commit data...
test_analyst_execution.py Loading commit data...
test_anthropic_effort.py Loading commit data...
test_api_key_env.py Loading commit data...
test_bangladesh_profile.py Loading commit data...
test_bedrock_provider.py Loading commit data...
test_capabilities.py Loading commit data...
test_checkpoint_resume.py Loading commit data...
test_cli_branding.py Loading commit data...
test_cli_config_precedence.py Loading commit data...
test_cli_env_skip.py Loading commit data...
test_cli_no_console.py Loading commit data...
test_cli_symbol_handling.py Loading commit data...
test_crypto_asset_mode.py Loading commit data...
test_dashboard_flow.py Loading commit data...
test_dataflows_config.py Loading commit data...
test_date_boundaries.py Loading commit data...
test_deepseek_reasoning.py Loading commit data...
test_dgt_forecasting.py Loading commit data...
test_dse_dataflow.py Loading commit data...
test_env_overrides.py Loading commit data...
test_fred.py Loading commit data...
test_google_api_key.py Loading commit data...
test_google_thinking_level.py Loading commit data...
test_i18n_coverage.py Loading commit data...
test_instrument_identity.py Loading commit data...
test_llm_max_retries.py Loading commit data...
test_market_data_validator.py Loading commit data...
test_market_toolnode.py Loading commit data...
test_memory_log.py Loading commit data...
test_minimax.py Loading commit data...
test_model_validation.py Loading commit data...
test_news_analyst_prompt.py Loading commit data...
test_news_lookahead.py Loading commit data...
test_no_data_handling.py Loading commit data...
test_ohlcv_cache_freshness.py Loading commit data...
test_ollama_base_url.py Loading commit data...
test_openai_compatible_provider.py Loading commit data...
test_openai_reasoning_effort.py Loading commit data...
test_openai_responses_base_url.py Loading commit data...
test_openrouter_body_retry.py Loading commit data...
test_openrouter_model_select.py Loading commit data...
test_polymarket.py Loading commit data...
test_provider_registry.py Loading commit data...
test_reddit_fallback.py Loading commit data...
test_reporting.py Loading commit data...
test_risk_router_path_map.py Loading commit data...
test_safe_ticker_component.py Loading commit data...
test_signal_processing.py Loading commit data...
test_stockstats_date_column.py Loading commit data...
test_stocktwits_resilience.py Loading commit data...
test_structured_agent_prompts.py Loading commit data...
test_structured_agents.py Loading commit data...
test_symbol_normalization_paths.py Loading commit data...
test_symbol_utils.py Loading commit data...
test_temperature_config.py Loading commit data...
test_ticker_symbol_handling.py Loading commit data...
test_timesfm_forecasting.py Loading commit data...
test_vendor_errors.py Loading commit data...
test_vendor_routing.py Loading commit data...
test_yfinance_stale_ohlcv_guard.py Loading commit data...