Hi Team! Last week, all my code using the neuralforecast library (specifically models PatchTST, NHITS and TimesNet) were all running fine. Today, without changing anything in the code, I am receiving an error when using

nf.fit(df=Y_train_df)

. The error is the following, do you guys have any idea what is going on here?

GPU available: True (cuda), used: True
TPU available: False, using: 0 TPU cores
IPU available: False, using: 0 IPUs
HPU available: False, using: 0 HPUs
LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]

  | Name         | Type          | Params
-----------------------------------------------
0 | loss         | MAE           | 0     
1 | padder_train | ConstantPad1d | 0     
2 | scaler       | TemporalNorm  | 0     
3 | blocks       | ModuleList    | 3.6 M 
-----------------------------------------------
3.6 M     Trainable params
0         Non-trainable params
3.6 M     Total params
14.229    Total estimated model params size (MB)
---------------------------------------------------------------------------
Exception                                 Traceback (most recent call last)
Cell In[14], line 1
----> 1 nf.fit(df=Y_train_df)

[...]

File /layers/paketo-buildpacks_cpython/cpython/lib/python3.10/site-packages/omegaconf/grammar_visitor.py:25
     22     from .base import Node  # noqa F401
     24 try:
---> 25     from omegaconf.grammar.gen.OmegaConfGrammarLexer import OmegaConfGrammarLexer
     26     from omegaconf.grammar.gen.OmegaConfGrammarParser import OmegaConfGrammarParser
     27     from omegaconf.grammar.gen.OmegaConfGrammarParserVisitor import (
     28         OmegaConfGrammarParserVisitor,
     29     )

File /layers/paketo-buildpacks_cpython/cpython/lib/python3.10/site-packages/omegaconf/grammar/gen/OmegaConfGrammarLexer.py:257
    253         buf.write("\3\2\t\f\2\t\r\2\t\24\2\t\4\2\t\25\2\t\32\2\t\33\2")
    254         return buf.getvalue()
--> 257 class OmegaConfGrammarLexer(Lexer):
    259     atn = ATNDeserializer().deserialize(serializedATN())
    261     decisionsToDFA = [ DFA(ds, i) for i, ds in enumerate(atn.decisionToState) ]

File /layers/paketo-buildpacks_cpython/cpython/lib/python3.10/site-packages/omegaconf/grammar/gen/OmegaConfGrammarLexer.py:259, in OmegaConfGrammarLexer()
    257 class OmegaConfGrammarLexer(Lexer):
--> 259     atn = ATNDeserializer().deserialize(serializedATN())
    261     decisionsToDFA = [ DFA(ds, i) for i, ds in enumerate(atn.decisionToState) ]
    263     VALUE_MODE = 1

File /layers/paketo-buildpacks_cpython/cpython/lib/python3.10/site-packages/antlr4/atn/ATNDeserializer.py:28, in ATNDeserializer.deserialize(self, data)
     26 self.data = data
     27 self.pos = 0
---> 28 self.checkVersion()
     29 atn = self.readATN()
     30 self.readStates(atn)

File /layers/paketo-buildpacks_cpython/cpython/lib/python3.10/site-packages/antlr4/atn/ATNDeserializer.py:50, in ATNDeserializer.checkVersion(self)
     48 version = self.readInt()
     49 if version != SERIALIZED_VERSION:
---> 50     raise Exception("Could not deserialize ATN with version " + str(version) + " (expected " + str(SERIALIZED_VERSION) + ").")

Exception: Could not deserialize ATN with version  (expected 4)

I'm specifically stumped with this, since all my code ran fine just last week and I didn't change anything. I have also tried just running a PatchTST and NHITS model from scratch using your tutorials, and I'm getting the same error. Even copying directly your code from the PatchTST usage example, gives this error for me. Any idea what is going on?

I was actually just typing that this solved the problem for me! Thanks for the reply, will I not run into future problems by just

pip uninstall omegaconf

?

Depends on if you're using it or not. I believe in that case it came as a dependency from somewhere else and wasn't being directly used

Amazing, thanks for all the help!