[BUG] Updated base_model.py to account for importing error

[Himanshu-Verma-ds]

changed the importing of pytorch-forecasting in order to account for this error "model must be a LightningModule or torch._dynamo.OptimizedModule, got TemporalFusionTransformer"

Description

This PR solves the issue of importing of pytorch-forecasting in order to account for this error "model must be a LightningModule or torch._dynamo.OptimizedModule, got TemporalFusionTransformer"

Checklist

• Linked issues (if existing)
• Amended changelog for large changes (and added myself there as contributor)
• Added/modified tests
• Used pre-commit hooks when committing to ensure that code is compliant with hooks. Install hooks with pre-commit install.
  To run hooks independent of commit, execute pre-commit run --all-files

Make sure to have fun coding!

[fnhirwa]

I think the current state of the repo fixed this issue, This should be closed.
FYI @fkiraly

[fkiraly]

Agree - though the PR would have fixed the issue as well, if merged.
I think we should just merge this PR (empty though) for sake of crediting that.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
⚠️ Please upload report for BASE (main@6ff532f). Learn more about missing BASE report.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1488   +/-   ##
=======================================
  Coverage        ?   86.36%           
=======================================
  Files           ?       96           
  Lines           ?     7801           
  Branches        ?        0           
=======================================
  Hits            ?     6737           
  Misses          ?     1064           
  Partials        ?        0           

Flag	Coverage Δ
cpu	86.36% <0.00%> (?)
pytest	86.36% <0.00%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ 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.