Update test_trading_envronment.py

kasinadhsarma · Aug 12, 2024 · c8e72ae · c8e72ae
1 parent db4d38c
commit c8e72ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_trading_envronment.py b/test/test_trading_envronment.py
@@ -2,7 +2,7 @@
 from unittest.mock import patch, MagicMock
 import numpy as np
 import gym
-from your_module import TradingEnvironment, moving_average_jax, load_data  # Replace 'your_module' with the actual module name
+from src.trading_environment import TradingEnvironment, moving_average_jax, load_data  # Replace 'your_module' with the actual module name
 
 class TestTradingEnvironment(unittest.TestCase):