Remove duplicated functions available in PMAT (#121)

agentcoinorg · Oct 31, 2024 · 8df988f · 8df988f
1 parent 7a7595c
commit 8df988f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 128 deletions.
diff --git a/prediction_prophet/benchmark/agents.py b/prediction_prophet/benchmark/agents.py
@@ -11,7 +11,6 @@
 from prediction_prophet.autonolas.research import EmbeddingModel
 from prediction_prophet.autonolas.research import make_prediction, get_urls_from_queries
 from prediction_prophet.autonolas.research import research as research_autonolas
-from prediction_prophet.functions.evaluate_question import is_predictable
 from prediction_prophet.functions.rephrase_question import rephrase_question
 from prediction_prophet.functions.research import Research, research as prophet_research
 from prediction_prophet.functions.search import search
@@ -26,6 +25,7 @@
 from pydantic.types import SecretStr
 from prediction_prophet.autonolas.research import Prediction as LLMCompletionPredictionDict
 from prediction_market_agent_tooling.tools.langfuse_ import observe
+from prediction_market_agent_tooling.tools.is_predictable import is_predictable_binary
 from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 
 if t.TYPE_CHECKING:
@@ -112,11 +112,11 @@ def __init__(
         self.embedding_model = embedding_model
 
     def is_predictable(self, market_question: str) -> bool:
-        (result, _) = is_predictable(question=market_question)
+        result = is_predictable_binary(question=market_question)
         return result
 
     def is_predictable_restricted(self, market_question: str, time_restriction_up_to: datetime) -> bool:
-        (result, _) = is_predictable(question=market_question)
+        result = is_predictable_binary(question=market_question)
         return result
 
     def research(self, market_question: str) -> str:
@@ -187,11 +187,11 @@ def __init__(
         self.logger = logger
 
     def is_predictable(self, market_question: str) -> bool:
-        (result, _) = is_predictable(question=market_question)
+        result = is_predictable_binary(question=market_question)
         return result
 
     def is_predictable_restricted(self, market_question: str, time_restriction_up_to: datetime) -> bool:
-        (result, _) = is_predictable(question=market_question)
+        result = is_predictable_binary(question=market_question)
         return result
 
     def research(self, market_question: str) -> Research:

diff --git a/prediction_prophet/functions/evaluate_question.py b/prediction_prophet/functions/evaluate_question.py
diff --git a/prediction_prophet/functions/is_predictable_and_binary.py b/prediction_prophet/functions/is_predictable_and_binary.py