From 2dcb8a5e184c9972c12339ee402e63dbcb68a123 Mon Sep 17 00:00:00 2001 From: Dobromir Popov Date: Wed, 30 Jul 2025 00:39:09 +0300 Subject: [PATCH] edit test --- test_massive_dqn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test_massive_dqn.py b/test_massive_dqn.py index 3d03c69..16b5e3a 100644 --- a/test_massive_dqn.py +++ b/test_massive_dqn.py @@ -46,12 +46,13 @@ def test_dqn_architecture(): output = network(test_input) if isinstance(output, tuple): - q_values, regime_pred, price_pred, volatility_pred, features = output + q_values, regime_pred, price_pred, volatility_pred, features, multi_timeframe_pred = output print(f" ✅ Q-values shape: {q_values.shape}") print(f" ✅ Regime prediction shape: {regime_pred.shape}") print(f" ✅ Price prediction shape: {price_pred.shape}") print(f" ✅ Volatility prediction shape: {volatility_pred.shape}") print(f" ✅ Features shape: {features.shape}") + print(f" ✅ Multi-timeframe predictions shape: {multi_timeframe_pred.shape}") else: print(f" ✅ Output shape: {output.shape}")