edsl 0.1.37.dev6__py3-none-any.whl → 0.1.38__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 (271) hide show
  1. edsl/Base.py +332 -303
  2. edsl/BaseDiff.py +260 -260
  3. edsl/TemplateLoader.py +24 -24
  4. edsl/__init__.py +49 -48
  5. edsl/__version__.py +1 -1
  6. edsl/agents/Agent.py +867 -855
  7. edsl/agents/AgentList.py +413 -350
  8. edsl/agents/Invigilator.py +233 -222
  9. edsl/agents/InvigilatorBase.py +265 -284
  10. edsl/agents/PromptConstructor.py +354 -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 -289
  26. edsl/config.py +157 -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 +1028 -958
  35. edsl/coop/utils.py +131 -131
  36. edsl/data/Cache.py +555 -527
  37. edsl/data/CacheEntry.py +233 -228
  38. edsl/data/CacheHandler.py +149 -149
  39. edsl/data/RemoteCacheSync.py +78 -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 +175 -173
  45. edsl/exceptions/BaseException.py +21 -21
  46. edsl/exceptions/__init__.py +54 -54
  47. edsl/exceptions/agents.py +42 -38
  48. edsl/exceptions/cache.py +5 -0
  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 +148 -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/PerplexityService.py +163 -0
  72. edsl/inference_services/TestService.py +89 -89
  73. edsl/inference_services/TogetherAIService.py +170 -170
  74. edsl/inference_services/models_available_cache.py +118 -118
  75. edsl/inference_services/rate_limits_cache.py +25 -25
  76. edsl/inference_services/registry.py +41 -39
  77. edsl/inference_services/write_available.py +10 -10
  78. edsl/jobs/Answers.py +56 -56
  79. edsl/jobs/Jobs.py +898 -1347
  80. edsl/jobs/JobsChecks.py +147 -0
  81. edsl/jobs/JobsPrompts.py +268 -0
  82. edsl/jobs/JobsRemoteInferenceHandler.py +239 -0
  83. edsl/jobs/__init__.py +1 -1
  84. edsl/jobs/buckets/BucketCollection.py +63 -63
  85. edsl/jobs/buckets/ModelBuckets.py +65 -65
  86. edsl/jobs/buckets/TokenBucket.py +251 -248
  87. edsl/jobs/interviews/Interview.py +661 -661
  88. edsl/jobs/interviews/InterviewExceptionCollection.py +99 -99
  89. edsl/jobs/interviews/InterviewExceptionEntry.py +186 -186
  90. edsl/jobs/interviews/InterviewStatistic.py +63 -63
  91. edsl/jobs/interviews/InterviewStatisticsCollection.py +25 -25
  92. edsl/jobs/interviews/InterviewStatusDictionary.py +78 -78
  93. edsl/jobs/interviews/InterviewStatusLog.py +92 -92
  94. edsl/jobs/interviews/ReportErrors.py +66 -66
  95. edsl/jobs/interviews/interview_status_enum.py +9 -9
  96. edsl/jobs/runners/JobsRunnerAsyncio.py +466 -338
  97. edsl/jobs/runners/JobsRunnerStatus.py +330 -332
  98. edsl/jobs/tasks/QuestionTaskCreator.py +242 -242
  99. edsl/jobs/tasks/TaskCreators.py +64 -64
  100. edsl/jobs/tasks/TaskHistory.py +450 -442
  101. edsl/jobs/tasks/TaskStatusLog.py +23 -23
  102. edsl/jobs/tasks/task_status_enum.py +163 -163
  103. edsl/jobs/tokens/InterviewTokenUsage.py +27 -27
  104. edsl/jobs/tokens/TokenUsage.py +34 -34
  105. edsl/language_models/KeyLookup.py +30 -30
  106. edsl/language_models/LanguageModel.py +668 -706
  107. edsl/language_models/ModelList.py +155 -102
  108. edsl/language_models/RegisterLanguageModelsMeta.py +184 -184
  109. edsl/language_models/__init__.py +3 -3
  110. edsl/language_models/fake_openai_call.py +15 -15
  111. edsl/language_models/fake_openai_service.py +61 -61
  112. edsl/language_models/registry.py +190 -137
  113. edsl/language_models/repair.py +156 -156
  114. edsl/language_models/unused/ReplicateBase.py +83 -83
  115. edsl/language_models/utilities.py +64 -64
  116. edsl/notebooks/Notebook.py +258 -259
  117. edsl/notebooks/__init__.py +1 -1
  118. edsl/prompts/Prompt.py +362 -357
  119. edsl/prompts/__init__.py +2 -2
  120. edsl/questions/AnswerValidatorMixin.py +289 -289
  121. edsl/questions/QuestionBase.py +664 -656
  122. edsl/questions/QuestionBaseGenMixin.py +161 -161
  123. edsl/questions/QuestionBasePromptsMixin.py +217 -234
  124. edsl/questions/QuestionBudget.py +227 -227
  125. edsl/questions/QuestionCheckBox.py +359 -359
  126. edsl/questions/QuestionExtract.py +182 -183
  127. edsl/questions/QuestionFreeText.py +114 -114
  128. edsl/questions/QuestionFunctional.py +166 -159
  129. edsl/questions/QuestionList.py +231 -231
  130. edsl/questions/QuestionMultipleChoice.py +286 -286
  131. edsl/questions/QuestionNumerical.py +153 -153
  132. edsl/questions/QuestionRank.py +324 -324
  133. edsl/questions/Quick.py +41 -41
  134. edsl/questions/RegisterQuestionsMeta.py +71 -71
  135. edsl/questions/ResponseValidatorABC.py +174 -174
  136. edsl/questions/SimpleAskMixin.py +73 -73
  137. edsl/questions/__init__.py +26 -26
  138. edsl/questions/compose_questions.py +98 -98
  139. edsl/questions/decorators.py +21 -21
  140. edsl/questions/derived/QuestionLikertFive.py +76 -76
  141. edsl/questions/derived/QuestionLinearScale.py +87 -87
  142. edsl/questions/derived/QuestionTopK.py +93 -91
  143. edsl/questions/derived/QuestionYesNo.py +82 -82
  144. edsl/questions/descriptors.py +413 -413
  145. edsl/questions/prompt_templates/question_budget.jinja +13 -13
  146. edsl/questions/prompt_templates/question_checkbox.jinja +32 -32
  147. edsl/questions/prompt_templates/question_extract.jinja +11 -11
  148. edsl/questions/prompt_templates/question_free_text.jinja +3 -3
  149. edsl/questions/prompt_templates/question_linear_scale.jinja +11 -11
  150. edsl/questions/prompt_templates/question_list.jinja +17 -17
  151. edsl/questions/prompt_templates/question_multiple_choice.jinja +33 -33
  152. edsl/questions/prompt_templates/question_numerical.jinja +36 -36
  153. edsl/questions/question_registry.py +177 -147
  154. edsl/questions/settings.py +12 -12
  155. edsl/questions/templates/budget/answering_instructions.jinja +7 -7
  156. edsl/questions/templates/budget/question_presentation.jinja +7 -7
  157. edsl/questions/templates/checkbox/answering_instructions.jinja +10 -10
  158. edsl/questions/templates/checkbox/question_presentation.jinja +22 -22
  159. edsl/questions/templates/extract/answering_instructions.jinja +7 -7
  160. edsl/questions/templates/likert_five/answering_instructions.jinja +10 -10
  161. edsl/questions/templates/likert_five/question_presentation.jinja +11 -11
  162. edsl/questions/templates/linear_scale/answering_instructions.jinja +5 -5
  163. edsl/questions/templates/linear_scale/question_presentation.jinja +5 -5
  164. edsl/questions/templates/list/answering_instructions.jinja +3 -3
  165. edsl/questions/templates/list/question_presentation.jinja +5 -5
  166. edsl/questions/templates/multiple_choice/answering_instructions.jinja +9 -9
  167. edsl/questions/templates/multiple_choice/question_presentation.jinja +11 -11
  168. edsl/questions/templates/numerical/answering_instructions.jinja +6 -6
  169. edsl/questions/templates/numerical/question_presentation.jinja +6 -6
  170. edsl/questions/templates/rank/answering_instructions.jinja +11 -11
  171. edsl/questions/templates/rank/question_presentation.jinja +15 -15
  172. edsl/questions/templates/top_k/answering_instructions.jinja +8 -8
  173. edsl/questions/templates/top_k/question_presentation.jinja +22 -22
  174. edsl/questions/templates/yes_no/answering_instructions.jinja +6 -6
  175. edsl/questions/templates/yes_no/question_presentation.jinja +11 -11
  176. edsl/results/CSSParameterizer.py +108 -0
  177. edsl/results/Dataset.py +424 -293
  178. edsl/results/DatasetExportMixin.py +731 -717
  179. edsl/results/DatasetTree.py +275 -145
  180. edsl/results/Result.py +465 -450
  181. edsl/results/Results.py +1165 -1071
  182. edsl/results/ResultsDBMixin.py +238 -238
  183. edsl/results/ResultsExportMixin.py +43 -43
  184. edsl/results/ResultsFetchMixin.py +33 -33
  185. edsl/results/ResultsGGMixin.py +121 -121
  186. edsl/results/ResultsToolsMixin.py +98 -98
  187. edsl/results/Selector.py +135 -135
  188. edsl/results/TableDisplay.py +198 -0
  189. edsl/results/__init__.py +2 -2
  190. edsl/results/table_display.css +78 -0
  191. edsl/results/tree_explore.py +115 -115
  192. edsl/scenarios/FileStore.py +632 -458
  193. edsl/scenarios/Scenario.py +601 -546
  194. edsl/scenarios/ScenarioHtmlMixin.py +64 -64
  195. edsl/scenarios/ScenarioJoin.py +127 -0
  196. edsl/scenarios/ScenarioList.py +1287 -1112
  197. edsl/scenarios/ScenarioListExportMixin.py +52 -52
  198. edsl/scenarios/ScenarioListPdfMixin.py +261 -261
  199. edsl/scenarios/__init__.py +4 -4
  200. edsl/shared.py +1 -1
  201. edsl/study/ObjectEntry.py +173 -173
  202. edsl/study/ProofOfWork.py +113 -113
  203. edsl/study/SnapShot.py +80 -80
  204. edsl/study/Study.py +528 -528
  205. edsl/study/__init__.py +4 -4
  206. edsl/surveys/DAG.py +148 -148
  207. edsl/surveys/Memory.py +31 -31
  208. edsl/surveys/MemoryPlan.py +244 -244
  209. edsl/surveys/Rule.py +326 -330
  210. edsl/surveys/RuleCollection.py +387 -387
  211. edsl/surveys/Survey.py +1801 -1795
  212. edsl/surveys/SurveyCSS.py +261 -261
  213. edsl/surveys/SurveyExportMixin.py +259 -259
  214. edsl/surveys/SurveyFlowVisualizationMixin.py +179 -121
  215. edsl/surveys/SurveyQualtricsImport.py +284 -284
  216. edsl/surveys/__init__.py +3 -3
  217. edsl/surveys/base.py +53 -53
  218. edsl/surveys/descriptors.py +56 -56
  219. edsl/surveys/instructions/ChangeInstruction.py +49 -47
  220. edsl/surveys/instructions/Instruction.py +65 -51
  221. edsl/surveys/instructions/InstructionCollection.py +77 -77
  222. edsl/templates/error_reporting/base.html +23 -23
  223. edsl/templates/error_reporting/exceptions_by_model.html +34 -34
  224. edsl/templates/error_reporting/exceptions_by_question_name.html +16 -16
  225. edsl/templates/error_reporting/exceptions_by_type.html +16 -16
  226. edsl/templates/error_reporting/interview_details.html +115 -115
  227. edsl/templates/error_reporting/interviews.html +19 -10
  228. edsl/templates/error_reporting/overview.html +4 -4
  229. edsl/templates/error_reporting/performance_plot.html +1 -1
  230. edsl/templates/error_reporting/report.css +73 -73
  231. edsl/templates/error_reporting/report.html +117 -117
  232. edsl/templates/error_reporting/report.js +25 -25
  233. edsl/tools/__init__.py +1 -1
  234. edsl/tools/clusters.py +192 -192
  235. edsl/tools/embeddings.py +27 -27
  236. edsl/tools/embeddings_plotting.py +118 -118
  237. edsl/tools/plotting.py +112 -112
  238. edsl/tools/summarize.py +18 -18
  239. edsl/utilities/SystemInfo.py +28 -28
  240. edsl/utilities/__init__.py +22 -22
  241. edsl/utilities/ast_utilities.py +25 -25
  242. edsl/utilities/data/Registry.py +6 -6
  243. edsl/utilities/data/__init__.py +1 -1
  244. edsl/utilities/data/scooter_results.json +1 -1
  245. edsl/utilities/decorators.py +77 -77
  246. edsl/utilities/gcp_bucket/cloud_storage.py +96 -96
  247. edsl/utilities/interface.py +627 -627
  248. edsl/{conjure → utilities}/naming_utilities.py +263 -263
  249. edsl/utilities/repair_functions.py +28 -28
  250. edsl/utilities/restricted_python.py +70 -70
  251. edsl/utilities/utilities.py +424 -409
  252. {edsl-0.1.37.dev6.dist-info → edsl-0.1.38.dist-info}/LICENSE +21 -21
  253. {edsl-0.1.37.dev6.dist-info → edsl-0.1.38.dist-info}/METADATA +2 -1
  254. edsl-0.1.38.dist-info/RECORD +277 -0
  255. edsl/conjure/AgentConstructionMixin.py +0 -160
  256. edsl/conjure/Conjure.py +0 -62
  257. edsl/conjure/InputData.py +0 -659
  258. edsl/conjure/InputDataCSV.py +0 -48
  259. edsl/conjure/InputDataMixinQuestionStats.py +0 -182
  260. edsl/conjure/InputDataPyRead.py +0 -91
  261. edsl/conjure/InputDataSPSS.py +0 -8
  262. edsl/conjure/InputDataStata.py +0 -8
  263. edsl/conjure/QuestionOptionMixin.py +0 -76
  264. edsl/conjure/QuestionTypeMixin.py +0 -23
  265. edsl/conjure/RawQuestion.py +0 -65
  266. edsl/conjure/SurveyResponses.py +0 -7
  267. edsl/conjure/__init__.py +0 -9
  268. edsl/conjure/examples/placeholder.txt +0 -0
  269. edsl/conjure/utilities.py +0 -201
  270. edsl-0.1.37.dev6.dist-info/RECORD +0 -283
  271. {edsl-0.1.37.dev6.dist-info → edsl-0.1.38.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,47 +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):
28
- return {
29
- "keep": self.keep,
30
- "drop": self.drop,
31
- "edsl_class_name": "ChangeInstruction",
32
- }
33
-
34
- @add_edsl_version
35
- def to_dict(self):
36
- return self._to_dict()
37
-
38
- def __hash__(self) -> int:
39
- """Return a hash of the question."""
40
- from edsl.utilities.utilities import dict_hash
41
-
42
- return dict_hash(self._to_dict())
43
-
44
- @classmethod
45
- @remove_edsl_version
46
- def from_dict(cls, data):
47
- 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,51 +1,65 @@
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):
22
- return {
23
- "name": self.name,
24
- "text": self.text,
25
- "edsl_class_name": "Instruction",
26
- "preamble": self.preamble,
27
- }
28
-
29
- def add_question(self, question) -> "Survey":
30
- from edsl import Survey
31
-
32
- return Survey([self, question])
33
-
34
- @add_edsl_version
35
- def to_dict(self):
36
- return self._to_dict()
37
-
38
- def __hash__(self) -> int:
39
- """Return a hash of the question."""
40
- from edsl.utilities.utilities import dict_hash
41
-
42
- return dict_hash(self._to_dict())
43
-
44
- @classmethod
45
- @remove_edsl_version
46
- def from_dict(cls, data):
47
- return cls(
48
- data["name"],
49
- data["text"],
50
- data.get("preamble", "You were given the following instructions:"),
51
- )
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 _repr_html_(self):
22
+ d = self.to_dict(add_edsl_version=False)
23
+ data = [[k, v] for k, v in d.items()]
24
+ from tabulate import tabulate
25
+
26
+ table = str(tabulate(data, headers=["keys", "values"], tablefmt="html"))
27
+ return f"<pre>{table}</pre>"
28
+
29
+ @classmethod
30
+ def example(cls) -> "Instruction":
31
+ return cls(name="example", text="This is an example instruction.")
32
+
33
+ def to_dict(self, add_edsl_version=True):
34
+ d = {
35
+ "name": self.name,
36
+ "text": self.text,
37
+ "edsl_class_name": "Instruction",
38
+ "preamble": self.preamble,
39
+ }
40
+ if add_edsl_version:
41
+ from edsl import __version__
42
+
43
+ d["edsl_version"] = __version__
44
+ d["edsl_class_name"] = "Instruction"
45
+ return d
46
+
47
+ def add_question(self, question) -> "Survey":
48
+ from edsl import Survey
49
+
50
+ return Survey([self, question])
51
+
52
+ def __hash__(self) -> int:
53
+ """Return a hash of the question."""
54
+ from edsl.utilities.utilities import dict_hash
55
+
56
+ return dict_hash(self.to_dict(add_edsl_version=False))
57
+
58
+ @classmethod
59
+ @remove_edsl_version
60
+ def from_dict(cls, data):
61
+ return cls(
62
+ data["name"],
63
+ data["text"],
64
+ data.get("preamble", "You were given the following instructions:"),
65
+ )
@@ -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>