Fixed error in type hints (#326)

* adding annonations import

* adding annonations import in more files

---------

Co-authored-by: Souryadip Das <[email protected]>

.pre-commit-config.yaml

@@ -29,7 +29,8 @@ repos:
         additional_dependencies:
           - flake8-future-import
         args:
-          - --select=FI
+          - --select=
+          - --ignore FI58
         entry: flake8
         files: '\.py$'
   - repo: https://github.com/pycqa/isort.git

prompt2model/dataset_processor/mock.py

@@ -1,4 +1,5 @@
 """A mock dataset processor for testing purposes."""
+from __future__ import annotations
 
 import datasets
 

prompt2model/model_executor/mock.py

@@ -1,4 +1,5 @@
 """A dummy class to generate model outputs (for testing purposes)."""
+from __future__ import annotations
 
 import datasets
 

prompt2model/model_retriever/base.py

@@ -1,4 +1,5 @@
 """An interface for model selection."""
+from __future__ import annotations
 
 from abc import ABC, abstractmethod
 

prompt2model/model_retriever/mock.py

@@ -1,4 +1,5 @@
 """An interface for model selection."""
+from __future__ import annotations
 
 from prompt2model.model_retriever import ModelRetriever
 from prompt2model.prompt_parser import PromptSpec

prompt2model/param_selector/mock.py

@@ -1,4 +1,5 @@
 """Mock model selector for testing purposes."""
+from __future__ import annotations
 
 from typing import Any
 

prompt2model/run_locally.py

@@ -1,4 +1,5 @@
 """A script to run the prompt2model pipeline locally."""
+from __future__ import annotations
 
 import argparse
 

prompt2model/utils/tevatron_utils/retrieve.py

@@ -1,4 +1,5 @@
 """Tools for doing efficient similarity search via the Tevatron/faiss libraries."""
+from __future__ import annotations
 
 import pickle