New Python Syntax Highlighting and TensorFlow/Transformers

Problem description:
When I create my model like this:

model = TFBertForSequenceClassification.from_pretrained(
    'bert-base-uncased',
    num_labels=3,
)

and then I call model.fit(), the syntax highlighting shows as “(function) fit: Unknown”

Steps to reproduce:

  • Create a Repl using the Python for data science template
  • Add the line import tensorflow as tf to import tensorflow
  • Install transformers from the Replit Package Manager
  • Add from transformers import TFBertForSequenceClassification
  • Create the model with:
model = TFBertForSequenceClassification.from_pretrained(
    'bert-base-uncased',
    num_labels=3,
)
  • Write model.fit(), and the syntax highlighter should not provide any details on this function. This is not the case with other editors/syntax highlighters.

Bug appears at this link:
https://replit.com/@CallumCM/LLM-Routing#main.py