code: def test_wp_high_k():
      results = nft_happy_wp.predict_mask(
          "Please pass the salt and [MASK]", top_k=3000
      )
      print(results)
  
      assert results[0].token == "pepper"
  
imports: from pytest import approx
  from happytransformer import HappyWordPrediction, WPTrainArgs, WPEvalArgs
  from happytransformer.happy_word_prediction import WordPredictionResult
  from tests.run_save_load import run_save_load
  import pytest
  from tests import happy_wp
label_desc: Use image node_id 2 for calling a global function with example usage: test_wp_high_k() without return types