-
Notifications
You must be signed in to change notification settings - Fork 685
[ENH] Test framework for ptf-v2
#1841
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Hi @fkiraly, @PranavBhatP, I tried adding |
The open issue is: how to do |
Hmmm - here is an idea: each model (intermediate base class) could have a For performance reasons, I would not carry out the check in usage, just in the test. For the metadata generated by D1, this is universal and can simply be tested by D1 tests - all D1 should produce the same metadata. |
pytorch_forecasting/models/temporal_fusion_transformer/tft_v2_metadata.py
Outdated
Show resolved
Hide resolved
Sorry I didn't quite understand what you mean here.
class _BasePtForecasterV2(_BasePtForecaster):
def _check_metadata():
if EncoderDecoderDataModule:
return test_enc_dec_dm()
elif TSLibDataModule:
return test_tslib_dm()
elif:
... # so on?
These are my initial thoughts, but I still need your help to understand what exactly is the idea, I'm a bit lost here😅 |
That is what I mean
... unless you add the |
Hi @fkiraly, do we need to add something else to this PR? We can add |
I think this is ready - the conflicts with #1886 need to be resolved though. |
done |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I made some small changes:
- deduplicate code copy-pasted twice in
_conftest
- why did you put the same code in two files? - revert change to
_BasePtForecaster
name, the name remains the same until we move to v2 - changed "metadata" filename to pkg
Description
This PR solves #1838
Implements Test framework for v2