address issue #50

nd-ball · Dec 21, 2023 · 74e7b76 · 74e7b76
1 parent 2417485
commit 74e7b76
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,11 +4,14 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/)
 and this project adheres to [Semantic Versioning](http://semver.org/).
 
+## [0.4.12] - 2023-12-21
+
+- Write item IDs to disk for train_and_evaluate. 
+
 ## [0.4.11] - 2023-12-21
 
 - Fix an issue with cli.evaluate.
 
-
 ## [0.4.10] - 2023-04-12
 
 - Fix an issue with codecov, and also allow for Python 3.10 and 3.11.

diff --git a/py_irt/cli.py b/py_irt/cli.py
@@ -233,7 +233,7 @@ def train_and_evaluate(
         outputs.append(
             {
                 "subject_id": dataset.observation_subjects[i],
-                # "example_id": dataset.observation_items[i],
+                "example_id": dataset.observation_items[i],
                 "response": dataset.observations[i],
                 "prediction": preds[i],
             }

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "py-irt"
-version = "0.4.11"
+version = "0.4.12"
 readme = "README.md"
 homepage = "https://github.com/nd-ball/py-irt/"
 description = "Bayesian IRT models in Python"