diff --git a/prompttools/experiment/__init__.py b/prompttools/experiment/__init__.py index f8420ed7..a477614a 100644 --- a/prompttools/experiment/__init__.py +++ b/prompttools/experiment/__init__.py @@ -5,9 +5,9 @@ # LICENSE file in the root directory of this source tree. -from experiment import Experiment -from openai_chat_experiment import OpenAIChatExperiment -from openai_completion_experiment import OpenAICompletionExperiment +from .experiment import Experiment +from .openai_chat_experiment import OpenAIChatExperiment +from .openai_completion_experiment import OpenAICompletionExperiment __all__ = ["OpenAIChatExperiment", "OpenAICompletionExperiment"] diff --git a/prompttools/experiment/experiment.py b/prompttools/experiment/experiment.py index 53a38489..1e948edb 100644 --- a/prompttools/experiment/experiment.py +++ b/prompttools/experiment/experiment.py @@ -13,10 +13,10 @@ import pandas as pd from prompttools.requests.request_queue import RequestQueue -from prompttools.experiment.widgets.feedback import FeedbackWidgetProvider -from prompttools.experiment.widgets.comparison import ComparisonWidgetProvider -from prompttools.experiment.widgets.utility import is_interactive -from prompttools.experiment.error import PromptExperimentException +from .widgets.feedback import FeedbackWidgetProvider +from .widgets.comparison import ComparisonWidgetProvider +from .widgets.utility import is_interactive +from .error import PromptExperimentException pd.set_option("display.max_colwidth", 0) diff --git a/prompttools/experiment/openai_chat_experiment.py b/prompttools/experiment/openai_chat_experiment.py index 6a7adc21..05adbcb9 100644 --- a/prompttools/experiment/openai_chat_experiment.py +++ b/prompttools/experiment/openai_chat_experiment.py @@ -11,7 +11,7 @@ import logging from prompttools.mock.mock import mock_chat_completion_fn -from prompttools.experiment import Experiment +from .experiment import Experiment class OpenAIChatExperiment(Experiment): diff --git a/prompttools/experiment/openai_completion_experiment.py b/prompttools/experiment/openai_completion_experiment.py index cd8ba096..50ddc5a5 100644 --- a/prompttools/experiment/openai_completion_experiment.py +++ b/prompttools/experiment/openai_completion_experiment.py @@ -10,7 +10,7 @@ import logging from prompttools.mock.mock import mock_completion_fn -from prompttools.experiment import Experiment +from .experiment import Experiment class OpenAICompletionExperiment(Experiment): diff --git a/prompttools/harness/__init__.py b/prompttools/harness/__init__.py index f97488df..8dbf5905 100644 --- a/prompttools/harness/__init__.py +++ b/prompttools/harness/__init__.py @@ -5,11 +5,11 @@ # LICENSE file in the root directory of this source tree. -from chat_history_harness import ChatHistoryExperimentationHarness -from chat_model_comparison_harness import ChatModelComparisonHarness -from harness import ExperimentationHarness -from prompt_template_harness import PromptTemplateExperimentationHarness -from system_prompt_harness import SystemPromptExperimentationHarness +from .harness import ExperimentationHarness +from .chat_history_harness import ChatHistoryExperimentationHarness +from .chat_model_comparison_harness import ChatModelComparisonHarness +from .prompt_template_harness import PromptTemplateExperimentationHarness +from .system_prompt_harness import SystemPromptExperimentationHarness __all__ = [ diff --git a/prompttools/harness/chat_history_harness.py b/prompttools/harness/chat_history_harness.py index 6f804c79..28742011 100644 --- a/prompttools/harness/chat_history_harness.py +++ b/prompttools/harness/chat_history_harness.py @@ -5,7 +5,7 @@ # LICENSE file in the root directory of this source tree. from typing import Dict, List, Optional -from prompttools.harness import ExperimentationHarness +from .harness import ExperimentationHarness from prompttools.experiment import OpenAIChatExperiment diff --git a/prompttools/harness/chat_model_comparison_harness.py b/prompttools/harness/chat_model_comparison_harness.py index 8a775f79..d95acd6f 100644 --- a/prompttools/harness/chat_model_comparison_harness.py +++ b/prompttools/harness/chat_model_comparison_harness.py @@ -5,7 +5,7 @@ # LICENSE file in the root directory of this source tree. from typing import Dict, List, Optional -from prompttools.harness import ExperimentationHarness +from .harness import ExperimentationHarness from prompttools.experiment import OpenAIChatExperiment diff --git a/prompttools/harness/prompt_template_harness.py b/prompttools/harness/prompt_template_harness.py index 6f2ad375..50a6684c 100644 --- a/prompttools/harness/prompt_template_harness.py +++ b/prompttools/harness/prompt_template_harness.py @@ -6,7 +6,7 @@ from typing import Dict, List, Optional import jinja2 -from prompttools.harness import ExperimentationHarness +from .harness import ExperimentationHarness from prompttools.experiment import OpenAICompletionExperiment diff --git a/prompttools/harness/system_prompt_harness.py b/prompttools/harness/system_prompt_harness.py index 5d1295fc..a91aa241 100644 --- a/prompttools/harness/system_prompt_harness.py +++ b/prompttools/harness/system_prompt_harness.py @@ -5,7 +5,7 @@ # LICENSE file in the root directory of this source tree. from typing import Dict, List, Optional -from prompttools.harness import ExperimentationHarness +from .harness import ExperimentationHarness from prompttools.experiment import OpenAIChatExperiment diff --git a/prompttools/version.py b/prompttools/version.py index 494a0d3f..78bcaa32 100644 --- a/prompttools/version.py +++ b/prompttools/version.py @@ -1,2 +1,2 @@ -__version__ = '0.0.9a0+3a42789' -git_version = '3a42789b80eea638f2f0616c433c97260c7976ca' +__version__ = '0.0.9a0+060c93f' +git_version = '060c93f633d3c787a1e59c4bb326b836a43c824d'