Fix TestDeepRepresentLinUCB

[alexnikulkov]

Summary:
The test was failing in CircleCI: https://app.circleci.com/pipelines/github/facebookresearch/ReAgent/2491/workflows/e153e4b9-795d-4c30-a3ff-6e4411b7dbf2/jobs/24796
I removed unnecessary parts which were breaking the test

Reviewed By: czxttkl

Differential Revision: D38874439

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D38874439

[codecov-commenter]

Codecov Report

Merging #672 (7dce006) into main (9085aae) will increase coverage by 83.92%.
The diff coverage is n/a.

@@            Coverage Diff             @@
##            main     #672       +/-   ##
==========================================
+ Coverage   3.44%   87.37%   +83.92%     
==========================================
  Files        356      362        +6     
  Lines      22748    23129      +381     
  Branches       0       44       +44     
==========================================
+ Hits         784    20208    +19424     
+ Misses     21964     2895    -19069     
- Partials       0       26       +26     

Impacted Files	Coverage Δ
...ent/test/training/cb/test_deep_represent_linucb.py	100.00% <ø> (+100.00%)	⬆️
...ng/src/main/scala/com/facebook/spark/rl/Udfs.scala	93.75% <0.00%> (ø)
reagent/test/test_data/ex_mdps.py	96.42% <0.00%> (ø)
.../src/main/scala/com/facebook/spark/rl/Helper.scala	66.66% <0.00%> (ø)
...rc/main/scala/com/facebook/spark/rl/Timeline.scala	84.41% <0.00%> (ø)
...cala/com/facebook/spark/rl/MultiStepTimeline.scala	92.72% <0.00%> (ø)
...c/main/scala/com/facebook/spark/rl/Constants.scala	100.00% <0.00%> (ø)
reagent/core/multiprocess_utils.py	23.07% <0.00%> (+23.07%)	⬆️
reagent/models/seq2slate_reward.py	25.56% <0.00%> (+25.56%)	⬆️
reagent/gym/envs/pomdp/state_embed_env.py	26.31% <0.00%> (+26.31%)	⬆️
... and 330 more

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.