From 74e7b760ee3702ea4bc1a5e6ab8cf1bf64822fb9 Mon Sep 17 00:00:00 2001
From: John Lalor <john.lalor@nd.edu>
Date: Thu, 21 Dec 2023 09:31:27 -0500
Subject: [PATCH] address issue #50

---
 CHANGELOG.md   | 5 ++++-
 py_irt/cli.py  | 2 +-
 pyproject.toml | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 4b3a64e..fc29806 100644
--- 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
index 1caf755..24a6fbc 100644
--- 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
index 10db4d7..a772e1f 100644
--- 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"