class_docstr: Checks that pre-trained Wizard models give the correct results.
class_docstr_tok: ["Checks","that","pre-trained","Wizard","models","give","the","correct","results","."]
code: def test_retrieval(self):
      _, test = testing_utils.eval_model(
          RETRIEVAL_OPTIONS, skip_valid=True
      )
      # for full dataset, remove `num_examples` from END2END_OPTIONS
      # self.assertAlmostEqual(test['accuracy'], 0.8631, places=4)
      # self.assertAlmostEqual(test['hits@5'], 0.9814, places=4)
      # self.assertAlmostEqual(test['hits@10'], 0.9917, places=4)
      self.assertAlmostEqual(test["accuracy"], 0.9141, places=4)
      self.assertAlmostEqual(test["hits@5"], 1.0, places=4)
      self.assertAlmostEqual(test["hits@10"], 1.0, places=4)
  
file_path: ParlAI/tests/nightly/gpu/test_wizard.py
imports: from parlai.core.agents import create_agent
  import unittest
  import parlai.scripts.display_data
  from parlai.tasks.wizard_of_wikipedia.agents import TOKEN_KNOWLEDGE
  import parlai.utils.testing
  from projects.wizard_of_wikipedia.knowledge_retriever.knowledge_retriever import KnowledgeRetrieverAgent
label_desc: Use image node_id 2 for calling the TestWizardModel obj's underlying member method code with example usage: obj.test_retrieval() without return types