I'm trying to retrieve historical feature values from Vertex AI Feature Store following the documentation. I'm running the example notebook from vertex-ai-samples. Everything works fine until running:

offline_store.fetch_historical_feature_values(entity_df=entity_df, features=[movies_feature])

which results to error:

BadRequest: 400 POST : Syntax error: Unexpected keyword PROTO at [23:22]

following the url:

{
  "error": {
    "code": 401,
    "message": "Request is missing required authentication credential. Expected OAuth 2 access token, login cookie or other valid authentication credential. See .",
    "errors": [
      {
        "message": "Login Required.",
        "domain": "global",
        "reason": "required",
        "location": "Authorization",
        "locationType": "header"
      }
    ],
    "status": "UNAUTHENTICATED",
    "details": [
      {
        "@type": "type.googleapis/google.rpc.ErrorInfo",
        "reason": "CREDENTIALS_MISSING",
        "domain": "googleapis",
        "metadata": {
          "service": "bigquery.googleapis",
          "method": "google.cloud.bigquery.v2.JobService.ListJobs"
        }
      }
    ]
  }
}

I have BigQuery Job User, BigQuery Resource Admin, BigQuery User and Colab Enterprise Admin roles. I have tried running the notebook both locally and in the Colab Enterprise but I'm getting the same error. In addition, I tried setting up a Service Account and providing the credentials in the function call in credentials parameter, but I'm getting the same error. Initially, I was running this in europe-west3, but switched to us-central1 with no success.

There's very little documentation about the usage of this functionality beside the sources attached and I'm baffled what could be causing this when everything else is working from the notebook correctly.

So, any idea what could be causing this?