Fix bug with context size in new nns

[Mr-Geekman]

Before submitting (must do checklist)

• Did you read the contribution guide?
• Did you update the docs? We use Numpy format for all the methods and classes.
• Did you write any new necessary tests?
• Did you update the CHANGELOG?

Proposed Changes

Fix bug like in #936.

Closing issues

[github-actions]

🚀 Deployed on https://deploy-preview-951--etna-docs.netlify.app

[codecov-commenter]

Codecov Report

❗ No coverage uploaded for pull request base (inference@d5eb926). Click here to learn what that means.
The diff coverage is n/a.

@@             Coverage Diff              @@
##             inference     #951   +/-   ##
============================================
  Coverage             ?   85.05%           
============================================
  Files                ?      132           
  Lines                ?     7580           
  Branches             ?        0           
============================================
  Hits                 ?     6447           
  Misses               ?     1133           
  Partials             ?        0           

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more