From c84c51434e32499ebea38bdba70c4920bb1369b7 Mon Sep 17 00:00:00 2001
From: jgrusewski <jgrusewski@localhost>
Date: Mon, 16 Mar 2026 09:25:52 +0100
Subject: [PATCH] fix(tests): add missing TradingState import to DQN trainer
 tests
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

The test module used `super::*` but TradingState lives in
crate::dqn, not in the trainer module — broke `cargo test --lib`.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 crates/ml/src/trainers/dqn/trainer/tests.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/crates/ml/src/trainers/dqn/trainer/tests.rs b/crates/ml/src/trainers/dqn/trainer/tests.rs
index 363361a7e..8bb396d5c 100644
--- a/crates/ml/src/trainers/dqn/trainer/tests.rs
+++ b/crates/ml/src/trainers/dqn/trainer/tests.rs
@@ -1,4 +1,5 @@
 use super::*;
+use crate::dqn::TradingState;
 use crate::hyperopt::ParameterSpace;
 use std::sync::OnceLock;