edsl 0.1.14__py3-none-any.whl → 0.1.40__py3-none-any.whl

edsl/TemplateLoader.py

@@ -0,0 +1,24 @@
+from importlib import resources
+from jinja2 import BaseLoader, TemplateNotFound
+import os
+
+
+class TemplateLoader(BaseLoader):
+    def __init__(self, package_name, templates_dir):
+        self.package_name = package_name
+        self.templates_dir = templates_dir
+
+    def get_source(self, environment, template):
+        try:
+            parts = [self.templates_dir] + template.split("/")
+            template_path = os.path.join(*parts)
+
+            # Use resources.files() to get a Traversable object
+            templates = resources.files(self.package_name).joinpath(self.templates_dir)
+
+            # Use the read_text() method of the Traversable object
+            content = templates.joinpath(template).read_text()
+
+            return content, None, lambda: True
+        except FileNotFoundError:
+            raise TemplateNotFound(template)

edsl/__init__.py

@@ -1,22 +1,58 @@
 import os
+import time
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 ROOT_DIR = os.path.dirname(BASE_DIR)
 
-from edsl.config import Config, CONFIG
+from edsl.__version__ import __version__
+
+# from edsl.config import Config, CONFIG
+
 from edsl.agents.Agent import Agent
 from edsl.agents.AgentList import AgentList
-from edsl.questions import (
-    QuestionFreeText,
-    QuestionMultipleChoice,
-    QuestionNumerical,
-    QuestionCheckBox,
-)
+
+from edsl.questions import QuestionBase
+from edsl.questions.question_registry import Question
+from edsl.questions import QuestionMultipleChoice
+from edsl.questions import QuestionCheckBox
+from edsl.questions import QuestionExtract
+from edsl.questions import QuestionFreeText
+from edsl.questions import QuestionFunctional
+from edsl.questions import QuestionLikertFive
+from edsl.questions import QuestionList
+from edsl.questions import QuestionMatrix
+from edsl.questions import QuestionDict
+from edsl.questions import QuestionLinearScale
+from edsl.questions import QuestionNumerical
+from edsl.questions import QuestionYesNo
+from edsl.questions import QuestionBudget
+from edsl.questions import QuestionRank
+from edsl.questions import QuestionTopK
+
 from edsl.scenarios.Scenario import Scenario
 from edsl.scenarios.ScenarioList import ScenarioList
+from edsl.scenarios.FileStore import FileStore
 
-from edsl.utilities.interface import print_dict_with_rich
+# from edsl.utilities.interface import print_dict_with_rich
 from edsl.surveys.Survey import Survey
+from edsl.language_models.model import Model
+from edsl.language_models.ModelList import ModelList
+
+from edsl.results.Results import Results
+from edsl.data.Cache import Cache
+
+# from edsl.data.CacheEntry import CacheEntry
+from edsl.data.CacheHandler import set_session_cache, unset_session_cache
+
+# from edsl.shared import shared_globals
+
+from edsl.jobs.Jobs import Jobs
+from edsl.notebooks.Notebook import Notebook
+
+# from edsl.study.Study import Study
+
+# from edsl.conjure.Conjure import Conjure
+from edsl.coop.coop import Coop
 
-from edsl.language_models.registry import Model
-from edsl.utilities.pastebin import post, get
+from edsl.surveys.instructions.Instruction import Instruction
+from edsl.surveys.instructions.ChangeInstruction import ChangeInstruction

edsl/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.1.40"