Fix bug causing network to be re-initialized!

[lebrice]

Signed-off-by: Fabrice Normandin [email protected]

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 52.94118% with 8 lines in your changes missing coverage. Please review.

Project coverage is 72.13%. Comparing base (e190484) to head (2e5967e).

Files with missing lines	Patch %	Lines
project/algorithms/jax_ppo_test.py	0.00%	5 Missing ⚠️
project/algorithms/text_classifier.py	40.00%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #132      +/-   ##
==========================================
- Coverage   72.14%   72.13%   -0.01%     
==========================================
  Files          50       50              
  Lines        3877     3890      +13     
==========================================
+ Hits         2797     2806       +9     
- Misses       1080     1084       +4     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.