edsl 0.1.38.dev2__py3-none-any.whl → 0.1.38.dev3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (248) hide show
  1. edsl/Base.py +303 -303
  2. edsl/BaseDiff.py +260 -260
  3. edsl/TemplateLoader.py +24 -24
  4. edsl/__init__.py +49 -49
  5. edsl/__version__.py +1 -1
  6. edsl/agents/Agent.py +858 -858
  7. edsl/agents/AgentList.py +362 -362
  8. edsl/agents/Invigilator.py +222 -222
  9. edsl/agents/InvigilatorBase.py +284 -284
  10. edsl/agents/PromptConstructor.py +353 -353
  11. edsl/agents/__init__.py +3 -3
  12. edsl/agents/descriptors.py +99 -99
  13. edsl/agents/prompt_helpers.py +129 -129
  14. edsl/auto/AutoStudy.py +117 -117
  15. edsl/auto/StageBase.py +230 -230
  16. edsl/auto/StageGenerateSurvey.py +178 -178
  17. edsl/auto/StageLabelQuestions.py +125 -125
  18. edsl/auto/StagePersona.py +61 -61
  19. edsl/auto/StagePersonaDimensionValueRanges.py +88 -88
  20. edsl/auto/StagePersonaDimensionValues.py +74 -74
  21. edsl/auto/StagePersonaDimensions.py +69 -69
  22. edsl/auto/StageQuestions.py +73 -73
  23. edsl/auto/SurveyCreatorPipeline.py +21 -21
  24. edsl/auto/utilities.py +224 -224
  25. edsl/base/Base.py +279 -279
  26. edsl/config.py +149 -149
  27. edsl/conversation/Conversation.py +290 -290
  28. edsl/conversation/car_buying.py +58 -58
  29. edsl/conversation/chips.py +95 -95
  30. edsl/conversation/mug_negotiation.py +81 -81
  31. edsl/conversation/next_speaker_utilities.py +93 -93
  32. edsl/coop/PriceFetcher.py +54 -54
  33. edsl/coop/__init__.py +2 -2
  34. edsl/coop/coop.py +961 -961
  35. edsl/coop/utils.py +131 -131
  36. edsl/data/Cache.py +530 -530
  37. edsl/data/CacheEntry.py +228 -228
  38. edsl/data/CacheHandler.py +149 -149
  39. edsl/data/RemoteCacheSync.py +97 -97
  40. edsl/data/SQLiteDict.py +292 -292
  41. edsl/data/__init__.py +4 -4
  42. edsl/data/orm.py +10 -10
  43. edsl/data_transfer_models.py +73 -73
  44. edsl/enums.py +173 -173
  45. edsl/exceptions/BaseException.py +21 -21
  46. edsl/exceptions/__init__.py +54 -54
  47. edsl/exceptions/agents.py +42 -42
  48. edsl/exceptions/cache.py +5 -5
  49. edsl/exceptions/configuration.py +16 -16
  50. edsl/exceptions/coop.py +10 -10
  51. edsl/exceptions/data.py +14 -14
  52. edsl/exceptions/general.py +34 -34
  53. edsl/exceptions/jobs.py +33 -33
  54. edsl/exceptions/language_models.py +63 -63
  55. edsl/exceptions/prompts.py +15 -15
  56. edsl/exceptions/questions.py +91 -91
  57. edsl/exceptions/results.py +29 -29
  58. edsl/exceptions/scenarios.py +22 -22
  59. edsl/exceptions/surveys.py +37 -37
  60. edsl/inference_services/AnthropicService.py +87 -87
  61. edsl/inference_services/AwsBedrock.py +120 -120
  62. edsl/inference_services/AzureAI.py +217 -217
  63. edsl/inference_services/DeepInfraService.py +18 -18
  64. edsl/inference_services/GoogleService.py +156 -156
  65. edsl/inference_services/GroqService.py +20 -20
  66. edsl/inference_services/InferenceServiceABC.py +147 -147
  67. edsl/inference_services/InferenceServicesCollection.py +97 -97
  68. edsl/inference_services/MistralAIService.py +123 -123
  69. edsl/inference_services/OllamaService.py +18 -18
  70. edsl/inference_services/OpenAIService.py +224 -224
  71. edsl/inference_services/TestService.py +89 -89
  72. edsl/inference_services/TogetherAIService.py +170 -170
  73. edsl/inference_services/models_available_cache.py +118 -118
  74. edsl/inference_services/rate_limits_cache.py +25 -25
  75. edsl/inference_services/registry.py +39 -39
  76. edsl/inference_services/write_available.py +10 -10
  77. edsl/jobs/Answers.py +56 -56
  78. edsl/jobs/Jobs.py +1358 -1358
  79. edsl/jobs/__init__.py +1 -1
  80. edsl/jobs/buckets/BucketCollection.py +63 -63
  81. edsl/jobs/buckets/ModelBuckets.py +65 -65
  82. edsl/jobs/buckets/TokenBucket.py +251 -251
  83. edsl/jobs/interviews/Interview.py +661 -661
  84. edsl/jobs/interviews/InterviewExceptionCollection.py +99 -99
  85. edsl/jobs/interviews/InterviewExceptionEntry.py +186 -186
  86. edsl/jobs/interviews/InterviewStatistic.py +63 -63
  87. edsl/jobs/interviews/InterviewStatisticsCollection.py +25 -25
  88. edsl/jobs/interviews/InterviewStatusDictionary.py +78 -78
  89. edsl/jobs/interviews/InterviewStatusLog.py +92 -92
  90. edsl/jobs/interviews/ReportErrors.py +66 -66
  91. edsl/jobs/interviews/interview_status_enum.py +9 -9
  92. edsl/jobs/runners/JobsRunnerAsyncio.py +361 -361
  93. edsl/jobs/runners/JobsRunnerStatus.py +332 -332
  94. edsl/jobs/tasks/QuestionTaskCreator.py +242 -242
  95. edsl/jobs/tasks/TaskCreators.py +64 -64
  96. edsl/jobs/tasks/TaskHistory.py +451 -451
  97. edsl/jobs/tasks/TaskStatusLog.py +23 -23
  98. edsl/jobs/tasks/task_status_enum.py +163 -163
  99. edsl/jobs/tokens/InterviewTokenUsage.py +27 -27
  100. edsl/jobs/tokens/TokenUsage.py +34 -34
  101. edsl/language_models/KeyLookup.py +30 -30
  102. edsl/language_models/LanguageModel.py +708 -708
  103. edsl/language_models/ModelList.py +109 -109
  104. edsl/language_models/RegisterLanguageModelsMeta.py +184 -184
  105. edsl/language_models/__init__.py +3 -3
  106. edsl/language_models/fake_openai_call.py +15 -15
  107. edsl/language_models/fake_openai_service.py +61 -61
  108. edsl/language_models/registry.py +137 -137
  109. edsl/language_models/repair.py +156 -156
  110. edsl/language_models/unused/ReplicateBase.py +83 -83
  111. edsl/language_models/utilities.py +64 -64
  112. edsl/notebooks/Notebook.py +258 -258
  113. edsl/notebooks/__init__.py +1 -1
  114. edsl/prompts/Prompt.py +357 -357
  115. edsl/prompts/__init__.py +2 -2
  116. edsl/questions/AnswerValidatorMixin.py +289 -289
  117. edsl/questions/QuestionBase.py +660 -660
  118. edsl/questions/QuestionBaseGenMixin.py +161 -161
  119. edsl/questions/QuestionBasePromptsMixin.py +217 -217
  120. edsl/questions/QuestionBudget.py +227 -227
  121. edsl/questions/QuestionCheckBox.py +359 -359
  122. edsl/questions/QuestionExtract.py +183 -183
  123. edsl/questions/QuestionFreeText.py +114 -114
  124. edsl/questions/QuestionFunctional.py +166 -166
  125. edsl/questions/QuestionList.py +231 -231
  126. edsl/questions/QuestionMultipleChoice.py +286 -286
  127. edsl/questions/QuestionNumerical.py +153 -153
  128. edsl/questions/QuestionRank.py +324 -324
  129. edsl/questions/Quick.py +41 -41
  130. edsl/questions/RegisterQuestionsMeta.py +71 -71
  131. edsl/questions/ResponseValidatorABC.py +174 -174
  132. edsl/questions/SimpleAskMixin.py +73 -73
  133. edsl/questions/__init__.py +26 -26
  134. edsl/questions/compose_questions.py +98 -98
  135. edsl/questions/decorators.py +21 -21
  136. edsl/questions/derived/QuestionLikertFive.py +76 -76
  137. edsl/questions/derived/QuestionLinearScale.py +87 -87
  138. edsl/questions/derived/QuestionTopK.py +93 -93
  139. edsl/questions/derived/QuestionYesNo.py +82 -82
  140. edsl/questions/descriptors.py +413 -413
  141. edsl/questions/prompt_templates/question_budget.jinja +13 -13
  142. edsl/questions/prompt_templates/question_checkbox.jinja +32 -32
  143. edsl/questions/prompt_templates/question_extract.jinja +11 -11
  144. edsl/questions/prompt_templates/question_free_text.jinja +3 -3
  145. edsl/questions/prompt_templates/question_linear_scale.jinja +11 -11
  146. edsl/questions/prompt_templates/question_list.jinja +17 -17
  147. edsl/questions/prompt_templates/question_multiple_choice.jinja +33 -33
  148. edsl/questions/prompt_templates/question_numerical.jinja +36 -36
  149. edsl/questions/question_registry.py +147 -147
  150. edsl/questions/settings.py +12 -12
  151. edsl/questions/templates/budget/answering_instructions.jinja +7 -7
  152. edsl/questions/templates/budget/question_presentation.jinja +7 -7
  153. edsl/questions/templates/checkbox/answering_instructions.jinja +10 -10
  154. edsl/questions/templates/checkbox/question_presentation.jinja +22 -22
  155. edsl/questions/templates/extract/answering_instructions.jinja +7 -7
  156. edsl/questions/templates/likert_five/answering_instructions.jinja +10 -10
  157. edsl/questions/templates/likert_five/question_presentation.jinja +11 -11
  158. edsl/questions/templates/linear_scale/answering_instructions.jinja +5 -5
  159. edsl/questions/templates/linear_scale/question_presentation.jinja +5 -5
  160. edsl/questions/templates/list/answering_instructions.jinja +3 -3
  161. edsl/questions/templates/list/question_presentation.jinja +5 -5
  162. edsl/questions/templates/multiple_choice/answering_instructions.jinja +9 -9
  163. edsl/questions/templates/multiple_choice/question_presentation.jinja +11 -11
  164. edsl/questions/templates/numerical/answering_instructions.jinja +6 -6
  165. edsl/questions/templates/numerical/question_presentation.jinja +6 -6
  166. edsl/questions/templates/rank/answering_instructions.jinja +11 -11
  167. edsl/questions/templates/rank/question_presentation.jinja +15 -15
  168. edsl/questions/templates/top_k/answering_instructions.jinja +8 -8
  169. edsl/questions/templates/top_k/question_presentation.jinja +22 -22
  170. edsl/questions/templates/yes_no/answering_instructions.jinja +6 -6
  171. edsl/questions/templates/yes_no/question_presentation.jinja +11 -11
  172. edsl/results/Dataset.py +293 -293
  173. edsl/results/DatasetExportMixin.py +717 -717
  174. edsl/results/DatasetTree.py +145 -145
  175. edsl/results/Result.py +456 -456
  176. edsl/results/Results.py +1071 -1071
  177. edsl/results/ResultsDBMixin.py +238 -238
  178. edsl/results/ResultsExportMixin.py +43 -43
  179. edsl/results/ResultsFetchMixin.py +33 -33
  180. edsl/results/ResultsGGMixin.py +121 -121
  181. edsl/results/ResultsToolsMixin.py +98 -98
  182. edsl/results/Selector.py +135 -135
  183. edsl/results/__init__.py +2 -2
  184. edsl/results/tree_explore.py +115 -115
  185. edsl/scenarios/FileStore.py +458 -458
  186. edsl/scenarios/Scenario.py +544 -544
  187. edsl/scenarios/ScenarioHtmlMixin.py +64 -64
  188. edsl/scenarios/ScenarioList.py +1112 -1112
  189. edsl/scenarios/ScenarioListExportMixin.py +52 -52
  190. edsl/scenarios/ScenarioListPdfMixin.py +261 -261
  191. edsl/scenarios/__init__.py +4 -4
  192. edsl/shared.py +1 -1
  193. edsl/study/ObjectEntry.py +173 -173
  194. edsl/study/ProofOfWork.py +113 -113
  195. edsl/study/SnapShot.py +80 -80
  196. edsl/study/Study.py +528 -528
  197. edsl/study/__init__.py +4 -4
  198. edsl/surveys/DAG.py +148 -148
  199. edsl/surveys/Memory.py +31 -31
  200. edsl/surveys/MemoryPlan.py +244 -244
  201. edsl/surveys/Rule.py +326 -326
  202. edsl/surveys/RuleCollection.py +387 -387
  203. edsl/surveys/Survey.py +1787 -1787
  204. edsl/surveys/SurveyCSS.py +261 -261
  205. edsl/surveys/SurveyExportMixin.py +259 -259
  206. edsl/surveys/SurveyFlowVisualizationMixin.py +121 -121
  207. edsl/surveys/SurveyQualtricsImport.py +284 -284
  208. edsl/surveys/__init__.py +3 -3
  209. edsl/surveys/base.py +53 -53
  210. edsl/surveys/descriptors.py +56 -56
  211. edsl/surveys/instructions/ChangeInstruction.py +49 -49
  212. edsl/surveys/instructions/Instruction.py +53 -53
  213. edsl/surveys/instructions/InstructionCollection.py +77 -77
  214. edsl/templates/error_reporting/base.html +23 -23
  215. edsl/templates/error_reporting/exceptions_by_model.html +34 -34
  216. edsl/templates/error_reporting/exceptions_by_question_name.html +16 -16
  217. edsl/templates/error_reporting/exceptions_by_type.html +16 -16
  218. edsl/templates/error_reporting/interview_details.html +115 -115
  219. edsl/templates/error_reporting/interviews.html +9 -9
  220. edsl/templates/error_reporting/overview.html +4 -4
  221. edsl/templates/error_reporting/performance_plot.html +1 -1
  222. edsl/templates/error_reporting/report.css +73 -73
  223. edsl/templates/error_reporting/report.html +117 -117
  224. edsl/templates/error_reporting/report.js +25 -25
  225. edsl/tools/__init__.py +1 -1
  226. edsl/tools/clusters.py +192 -192
  227. edsl/tools/embeddings.py +27 -27
  228. edsl/tools/embeddings_plotting.py +118 -118
  229. edsl/tools/plotting.py +112 -112
  230. edsl/tools/summarize.py +18 -18
  231. edsl/utilities/SystemInfo.py +28 -28
  232. edsl/utilities/__init__.py +22 -22
  233. edsl/utilities/ast_utilities.py +25 -25
  234. edsl/utilities/data/Registry.py +6 -6
  235. edsl/utilities/data/__init__.py +1 -1
  236. edsl/utilities/data/scooter_results.json +1 -1
  237. edsl/utilities/decorators.py +77 -77
  238. edsl/utilities/gcp_bucket/cloud_storage.py +96 -96
  239. edsl/utilities/interface.py +627 -627
  240. edsl/utilities/naming_utilities.py +263 -263
  241. edsl/utilities/repair_functions.py +28 -28
  242. edsl/utilities/restricted_python.py +70 -70
  243. edsl/utilities/utilities.py +409 -409
  244. {edsl-0.1.38.dev2.dist-info → edsl-0.1.38.dev3.dist-info}/LICENSE +21 -21
  245. {edsl-0.1.38.dev2.dist-info → edsl-0.1.38.dev3.dist-info}/METADATA +1 -1
  246. edsl-0.1.38.dev3.dist-info/RECORD +269 -0
  247. edsl-0.1.38.dev2.dist-info/RECORD +0 -269
  248. {edsl-0.1.38.dev2.dist-info → edsl-0.1.38.dev3.dist-info}/WHEEL +0 -0
edsl/surveys/base.py CHANGED
@@ -1,53 +1,53 @@
1
- """Base classes and enumerations for the surveyor package."""
2
-
3
- from enum import Enum
4
-
5
-
6
- class RulePriority(Enum):
7
- """Enumeration of the priority of a rule."""
8
-
9
- DEFAULT = -1
10
-
11
-
12
- class EndOfSurveyParent:
13
- """A named object that represents the end of the survey."""
14
-
15
- pass
16
-
17
- def __repr__(self):
18
- """Return a string representation of the object."""
19
- return "EndOfSurvey"
20
-
21
- def __str__(self):
22
- """Return a string representation of the object."""
23
- return "EndOfSurvey"
24
-
25
- def __bool__(self):
26
- """Return False."""
27
- return False
28
-
29
- def __add__(self, other):
30
- """Add the object to another object.
31
-
32
- Example:
33
- >>> e = EndOfSurveyParent()
34
- >>> e + 1
35
- EndOfSurvey
36
- """
37
- return self
38
-
39
- def __deepcopy__(self, memo):
40
- # Return the same instance when deepcopy is called
41
- return self
42
-
43
- def __radd__(self, other):
44
- """Add the object to another object.
45
-
46
- Example:
47
- >>> 1 + EndOfSurveyParent()
48
- EndOfSurvey
49
- """
50
- return self
51
-
52
-
53
- EndOfSurvey = EndOfSurveyParent()
1
+ """Base classes and enumerations for the surveyor package."""
2
+
3
+ from enum import Enum
4
+
5
+
6
+ class RulePriority(Enum):
7
+ """Enumeration of the priority of a rule."""
8
+
9
+ DEFAULT = -1
10
+
11
+
12
+ class EndOfSurveyParent:
13
+ """A named object that represents the end of the survey."""
14
+
15
+ pass
16
+
17
+ def __repr__(self):
18
+ """Return a string representation of the object."""
19
+ return "EndOfSurvey"
20
+
21
+ def __str__(self):
22
+ """Return a string representation of the object."""
23
+ return "EndOfSurvey"
24
+
25
+ def __bool__(self):
26
+ """Return False."""
27
+ return False
28
+
29
+ def __add__(self, other):
30
+ """Add the object to another object.
31
+
32
+ Example:
33
+ >>> e = EndOfSurveyParent()
34
+ >>> e + 1
35
+ EndOfSurvey
36
+ """
37
+ return self
38
+
39
+ def __deepcopy__(self, memo):
40
+ # Return the same instance when deepcopy is called
41
+ return self
42
+
43
+ def __radd__(self, other):
44
+ """Add the object to another object.
45
+
46
+ Example:
47
+ >>> 1 + EndOfSurveyParent()
48
+ EndOfSurvey
49
+ """
50
+ return self
51
+
52
+
53
+ EndOfSurvey = EndOfSurveyParent()
@@ -1,56 +1,56 @@
1
- """This module contains the descriptors for the classes in the edsl package."""
2
-
3
- from abc import ABC, abstractmethod
4
- from typing import Any
5
- from edsl.questions.QuestionBase import QuestionBase
6
-
7
-
8
- class BaseDescriptor(ABC):
9
- """ABC for something."""
10
-
11
- @abstractmethod
12
- def validate(self, value: Any) -> None:
13
- """Validate the value. If it is invalid, raise an exception. If it is valid, do nothing."""
14
- pass
15
-
16
- def __get__(self, instance, owner):
17
- """Get the value of the attribute."""
18
- return instance.__dict__[self.name]
19
-
20
- def __set__(self, instance, value: Any) -> None:
21
- """Set the value of the attribute."""
22
- self.validate(value, instance)
23
- instance.__dict__[self.name] = value
24
-
25
- def __set_name__(self, owner, name: str) -> None:
26
- """Set the name of the attribute."""
27
- self.name = "_" + name
28
-
29
-
30
- class QuestionsDescriptor(BaseDescriptor):
31
- """Descriptor for questions."""
32
-
33
- def __get__(self, instance, owner):
34
- """Get the value of the attribute."""
35
- return instance.__dict__[self.name]
36
-
37
- def validate(self, value: Any, instance) -> None:
38
- """Validate the value. If it is invalid, raise an exception. If it is valid, do nothing."""
39
- if not isinstance(value, list):
40
- raise TypeError("Questions must be a list.")
41
- if not all(isinstance(question, QuestionBase) for question in value):
42
- raise TypeError("Questions must be a list of Question objects.")
43
- question_names = [question.question_name for question in value]
44
- if len(question_names) != len(set(question_names)):
45
- raise ValueError("Question names must be unique.")
46
-
47
- def __set__(self, instance, value: Any) -> None:
48
- """Set the value of the attribute."""
49
- self.validate(value, instance)
50
- instance.__dict__[self.name] = []
51
- for question in value:
52
- instance.add_question(question)
53
-
54
- def __set_name__(self, owner, name: str) -> None:
55
- """Set the name of the attribute."""
56
- self.name = "_" + name
1
+ """This module contains the descriptors for the classes in the edsl package."""
2
+
3
+ from abc import ABC, abstractmethod
4
+ from typing import Any
5
+ from edsl.questions.QuestionBase import QuestionBase
6
+
7
+
8
+ class BaseDescriptor(ABC):
9
+ """ABC for something."""
10
+
11
+ @abstractmethod
12
+ def validate(self, value: Any) -> None:
13
+ """Validate the value. If it is invalid, raise an exception. If it is valid, do nothing."""
14
+ pass
15
+
16
+ def __get__(self, instance, owner):
17
+ """Get the value of the attribute."""
18
+ return instance.__dict__[self.name]
19
+
20
+ def __set__(self, instance, value: Any) -> None:
21
+ """Set the value of the attribute."""
22
+ self.validate(value, instance)
23
+ instance.__dict__[self.name] = value
24
+
25
+ def __set_name__(self, owner, name: str) -> None:
26
+ """Set the name of the attribute."""
27
+ self.name = "_" + name
28
+
29
+
30
+ class QuestionsDescriptor(BaseDescriptor):
31
+ """Descriptor for questions."""
32
+
33
+ def __get__(self, instance, owner):
34
+ """Get the value of the attribute."""
35
+ return instance.__dict__[self.name]
36
+
37
+ def validate(self, value: Any, instance) -> None:
38
+ """Validate the value. If it is invalid, raise an exception. If it is valid, do nothing."""
39
+ if not isinstance(value, list):
40
+ raise TypeError("Questions must be a list.")
41
+ if not all(isinstance(question, QuestionBase) for question in value):
42
+ raise TypeError("Questions must be a list of Question objects.")
43
+ question_names = [question.question_name for question in value]
44
+ if len(question_names) != len(set(question_names)):
45
+ raise ValueError("Question names must be unique.")
46
+
47
+ def __set__(self, instance, value: Any) -> None:
48
+ """Set the value of the attribute."""
49
+ self.validate(value, instance)
50
+ instance.__dict__[self.name] = []
51
+ for question in value:
52
+ instance.add_question(question)
53
+
54
+ def __set_name__(self, owner, name: str) -> None:
55
+ """Set the name of the attribute."""
56
+ self.name = "_" + name
@@ -1,49 +1,49 @@
1
- from typing import List, Optional
2
-
3
- from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
4
-
5
-
6
- class ChangeInstruction:
7
- def __init__(
8
- self,
9
- keep: Optional[List[str]] = None,
10
- drop: Optional[List[str]] = None,
11
- ):
12
- if keep is None and drop is None:
13
- raise ValueError("Keep and drop cannot both be None")
14
-
15
- self.keep = keep or []
16
- self.drop = drop or []
17
-
18
- def include_instruction(self, instruction_name) -> bool:
19
- return (instruction_name in self.keep) or (not instruction_name in self.drop)
20
-
21
- def add_name(self, index) -> None:
22
- self.name = "change_instruction_{}".format(index)
23
-
24
- def __str__(self):
25
- return self.text
26
-
27
- def to_dict(self, add_edsl_version=True):
28
- d = {
29
- "keep": self.keep,
30
- "drop": self.drop,
31
- }
32
- if add_edsl_version:
33
- from edsl import __version__
34
-
35
- d["edsl_version"] = __version__
36
- d["edsl_class_name"] = "ChangeInstruction"
37
-
38
- return d
39
-
40
- def __hash__(self) -> int:
41
- """Return a hash of the question."""
42
- from edsl.utilities.utilities import dict_hash
43
-
44
- return dict_hash(self.to_dict(add_edsl_version=False))
45
-
46
- @classmethod
47
- @remove_edsl_version
48
- def from_dict(cls, data):
49
- return cls(data["keep"], data["drop"])
1
+ from typing import List, Optional
2
+
3
+ from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
4
+
5
+
6
+ class ChangeInstruction:
7
+ def __init__(
8
+ self,
9
+ keep: Optional[List[str]] = None,
10
+ drop: Optional[List[str]] = None,
11
+ ):
12
+ if keep is None and drop is None:
13
+ raise ValueError("Keep and drop cannot both be None")
14
+
15
+ self.keep = keep or []
16
+ self.drop = drop or []
17
+
18
+ def include_instruction(self, instruction_name) -> bool:
19
+ return (instruction_name in self.keep) or (not instruction_name in self.drop)
20
+
21
+ def add_name(self, index) -> None:
22
+ self.name = "change_instruction_{}".format(index)
23
+
24
+ def __str__(self):
25
+ return self.text
26
+
27
+ def to_dict(self, add_edsl_version=True):
28
+ d = {
29
+ "keep": self.keep,
30
+ "drop": self.drop,
31
+ }
32
+ if add_edsl_version:
33
+ from edsl import __version__
34
+
35
+ d["edsl_version"] = __version__
36
+ d["edsl_class_name"] = "ChangeInstruction"
37
+
38
+ return d
39
+
40
+ def __hash__(self) -> int:
41
+ """Return a hash of the question."""
42
+ from edsl.utilities.utilities import dict_hash
43
+
44
+ return dict_hash(self.to_dict(add_edsl_version=False))
45
+
46
+ @classmethod
47
+ @remove_edsl_version
48
+ def from_dict(cls, data):
49
+ return cls(data["keep"], data["drop"])
@@ -1,53 +1,53 @@
1
- from typing import Union, Optional, List, Generator, Dict
2
- from edsl.questions import QuestionBase
3
-
4
- from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
5
-
6
-
7
- class Instruction:
8
- def __init__(
9
- self, name, text, preamble="You were given the following instructions:"
10
- ):
11
- self.name = name
12
- self.text = text
13
- self.preamble = preamble
14
-
15
- def __str__(self):
16
- return self.text
17
-
18
- def __repr__(self):
19
- return """Instruction(name="{}", text="{}")""".format(self.name, self.text)
20
-
21
- def to_dict(self, add_edsl_version=True):
22
- d = {
23
- "name": self.name,
24
- "text": self.text,
25
- "edsl_class_name": "Instruction",
26
- "preamble": self.preamble,
27
- }
28
- if add_edsl_version:
29
- from edsl import __version__
30
-
31
- d["edsl_version"] = __version__
32
- d["edsl_class_name"] = "Instruction"
33
- return d
34
-
35
- def add_question(self, question) -> "Survey":
36
- from edsl import Survey
37
-
38
- return Survey([self, question])
39
-
40
- def __hash__(self) -> int:
41
- """Return a hash of the question."""
42
- from edsl.utilities.utilities import dict_hash
43
-
44
- return dict_hash(self.to_dict(add_edsl_version=False))
45
-
46
- @classmethod
47
- @remove_edsl_version
48
- def from_dict(cls, data):
49
- return cls(
50
- data["name"],
51
- data["text"],
52
- data.get("preamble", "You were given the following instructions:"),
53
- )
1
+ from typing import Union, Optional, List, Generator, Dict
2
+ from edsl.questions import QuestionBase
3
+
4
+ from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
5
+
6
+
7
+ class Instruction:
8
+ def __init__(
9
+ self, name, text, preamble="You were given the following instructions:"
10
+ ):
11
+ self.name = name
12
+ self.text = text
13
+ self.preamble = preamble
14
+
15
+ def __str__(self):
16
+ return self.text
17
+
18
+ def __repr__(self):
19
+ return """Instruction(name="{}", text="{}")""".format(self.name, self.text)
20
+
21
+ def to_dict(self, add_edsl_version=True):
22
+ d = {
23
+ "name": self.name,
24
+ "text": self.text,
25
+ "edsl_class_name": "Instruction",
26
+ "preamble": self.preamble,
27
+ }
28
+ if add_edsl_version:
29
+ from edsl import __version__
30
+
31
+ d["edsl_version"] = __version__
32
+ d["edsl_class_name"] = "Instruction"
33
+ return d
34
+
35
+ def add_question(self, question) -> "Survey":
36
+ from edsl import Survey
37
+
38
+ return Survey([self, question])
39
+
40
+ def __hash__(self) -> int:
41
+ """Return a hash of the question."""
42
+ from edsl.utilities.utilities import dict_hash
43
+
44
+ return dict_hash(self.to_dict(add_edsl_version=False))
45
+
46
+ @classmethod
47
+ @remove_edsl_version
48
+ def from_dict(cls, data):
49
+ return cls(
50
+ data["name"],
51
+ data["text"],
52
+ data.get("preamble", "You were given the following instructions:"),
53
+ )
@@ -1,77 +1,77 @@
1
- from edsl.surveys.instructions.Instruction import Instruction
2
- from edsl.surveys.instructions.ChangeInstruction import ChangeInstruction
3
- from edsl.questions import QuestionBase
4
- from typing import Union, Optional, List, Generator, Dict
5
-
6
-
7
- from collections import UserDict
8
-
9
-
10
- class InstructionCollection(UserDict):
11
- def __init__(
12
- self,
13
- instruction_names_to_instruction: Dict[str, Instruction],
14
- questions: List[QuestionBase],
15
- ):
16
- self.instruction_names_to_instruction = instruction_names_to_instruction
17
- self.questions = questions
18
- data = {}
19
- for question in self.questions:
20
- # just add both the question and the question name
21
- data[question.name] = list(self.relevant_instructions(question))
22
- # data[question] = list(self.relevant_instructions(question))
23
- super().__init__(data)
24
-
25
- def __getitem__(self, key):
26
- # in case the person uses question instead of the name
27
- if isinstance(key, QuestionBase):
28
- key = key.name
29
- return self.data[key]
30
-
31
- @property
32
- def question_names(self):
33
- return [q.name for q in self.questions]
34
-
35
- def question_index(self, question_name):
36
- return self.question_names.index(question_name)
37
-
38
- def _entries_before(
39
- self, question_name
40
- ) -> tuple[List[Instruction], List[ChangeInstruction]]:
41
- if question_name not in self.question_names:
42
- raise ValueError(
43
- f"Question name not found in the list of questions: got '{question_name}'; list is {self.question_names}"
44
- )
45
- instructions, changes = [], []
46
-
47
- index = self.question_index(question_name)
48
- for instruction in self.instruction_names_to_instruction.values():
49
- if instruction.pseudo_index < index:
50
- if isinstance(instruction, Instruction):
51
- instructions.append(instruction)
52
- elif isinstance(instruction, ChangeInstruction):
53
- changes.append(instruction)
54
- return instructions, changes
55
-
56
- def relevant_instructions(
57
- self, question: Union[str, QuestionBase]
58
- ) -> Generator[Instruction, None, None]:
59
- ## Find all the questions that are after a given instruction
60
- if isinstance(question, str):
61
- question_name = question
62
- elif isinstance(question, QuestionBase):
63
- question_name = question.name
64
-
65
- instructions_before, changes_before = self._entries_before(question_name)
66
- keep_list = []
67
- drop_list = []
68
- for change in changes_before:
69
- keep_list.extend(change.keep)
70
- drop_list.extend(change.drop)
71
-
72
- for instruction in instructions_before:
73
- if instruction.name in keep_list or instruction.name not in drop_list:
74
- yield instruction
75
-
76
- def __len__(self):
77
- return len(self.instruction_names_to_instruction)
1
+ from edsl.surveys.instructions.Instruction import Instruction
2
+ from edsl.surveys.instructions.ChangeInstruction import ChangeInstruction
3
+ from edsl.questions import QuestionBase
4
+ from typing import Union, Optional, List, Generator, Dict
5
+
6
+
7
+ from collections import UserDict
8
+
9
+
10
+ class InstructionCollection(UserDict):
11
+ def __init__(
12
+ self,
13
+ instruction_names_to_instruction: Dict[str, Instruction],
14
+ questions: List[QuestionBase],
15
+ ):
16
+ self.instruction_names_to_instruction = instruction_names_to_instruction
17
+ self.questions = questions
18
+ data = {}
19
+ for question in self.questions:
20
+ # just add both the question and the question name
21
+ data[question.name] = list(self.relevant_instructions(question))
22
+ # data[question] = list(self.relevant_instructions(question))
23
+ super().__init__(data)
24
+
25
+ def __getitem__(self, key):
26
+ # in case the person uses question instead of the name
27
+ if isinstance(key, QuestionBase):
28
+ key = key.name
29
+ return self.data[key]
30
+
31
+ @property
32
+ def question_names(self):
33
+ return [q.name for q in self.questions]
34
+
35
+ def question_index(self, question_name):
36
+ return self.question_names.index(question_name)
37
+
38
+ def _entries_before(
39
+ self, question_name
40
+ ) -> tuple[List[Instruction], List[ChangeInstruction]]:
41
+ if question_name not in self.question_names:
42
+ raise ValueError(
43
+ f"Question name not found in the list of questions: got '{question_name}'; list is {self.question_names}"
44
+ )
45
+ instructions, changes = [], []
46
+
47
+ index = self.question_index(question_name)
48
+ for instruction in self.instruction_names_to_instruction.values():
49
+ if instruction.pseudo_index < index:
50
+ if isinstance(instruction, Instruction):
51
+ instructions.append(instruction)
52
+ elif isinstance(instruction, ChangeInstruction):
53
+ changes.append(instruction)
54
+ return instructions, changes
55
+
56
+ def relevant_instructions(
57
+ self, question: Union[str, QuestionBase]
58
+ ) -> Generator[Instruction, None, None]:
59
+ ## Find all the questions that are after a given instruction
60
+ if isinstance(question, str):
61
+ question_name = question
62
+ elif isinstance(question, QuestionBase):
63
+ question_name = question.name
64
+
65
+ instructions_before, changes_before = self._entries_before(question_name)
66
+ keep_list = []
67
+ drop_list = []
68
+ for change in changes_before:
69
+ keep_list.extend(change.keep)
70
+ drop_list.extend(change.drop)
71
+
72
+ for instruction in instructions_before:
73
+ if instruction.name in keep_list or instruction.name not in drop_list:
74
+ yield instruction
75
+
76
+ def __len__(self):
77
+ return len(self.instruction_names_to_instruction)
@@ -1,24 +1,24 @@
1
- <!DOCTYPE html>
2
- <html lang="en">
3
- <head>
4
- <meta charset="UTF-8">
5
- <meta name="viewport" content="width=device-width, initial-scale=1.0">
6
- <title>Exception Details</title>
7
- <style>
8
- {{ css }}
9
- </style>
10
-
11
- <script>
12
- {{ javascript }}
13
- </script>
14
-
15
- </head>
16
- <body>
17
- {% include 'overview.html' %}
18
- {% include 'exceptions_by_type.html' %}
19
- {% include 'exceptions_by_model.html' %}
20
- {% include 'exceptions_by_question_name.html' %}
21
- {% include 'interviews.html' %}
22
- {% include 'performance_plot.html' %}
23
- </body>
1
+ <!DOCTYPE html>
2
+ <html lang="en">
3
+ <head>
4
+ <meta charset="UTF-8">
5
+ <meta name="viewport" content="width=device-width, initial-scale=1.0">
6
+ <title>Exception Details</title>
7
+ <style>
8
+ {{ css }}
9
+ </style>
10
+
11
+ <script>
12
+ {{ javascript }}
13
+ </script>
14
+
15
+ </head>
16
+ <body>
17
+ {% include 'overview.html' %}
18
+ {% include 'exceptions_by_type.html' %}
19
+ {% include 'exceptions_by_model.html' %}
20
+ {% include 'exceptions_by_question_name.html' %}
21
+ {% include 'interviews.html' %}
22
+ {% include 'performance_plot.html' %}
23
+ </body>
24
24
  </html>