From 2312af4085e04a0a76408c4961ac33da0bce14cd Mon Sep 17 00:00:00 2001
From: Nikhil Woodruff <nikhil.woodruff@outlook.com>
Date: Mon, 2 Jun 2025 08:44:51 +0100
Subject: [PATCH] Remove type check for `Dataset` class Fixes #157

---
 changelog_entry.yaml       | 4 ++++
 policyengine/simulation.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/changelog_entry.yaml b/changelog_entry.yaml
index e69de29b..3ab866b0 100644
--- a/changelog_entry.yaml
+++ b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    fixed:
+    - Removed bad type check.
diff --git a/policyengine/simulation.py b/policyengine/simulation.py
index a8a818fb..551bae3a 100644
--- a/policyengine/simulation.py
+++ b/policyengine/simulation.py
@@ -37,7 +37,7 @@
 CountryType = Literal["uk", "us"]
 ScopeType = Literal["household", "macro"]
 DataType = (
-    str | dict[Any, Any] | Dataset | None
+    str | dict[Any, Any] | Any | None
 )  # Needs stricter typing. Any==policyengine_core.data.Dataset, but pydantic refuses for some reason.
 TimePeriodType = int
 ReformType = ParametricReform | Type[StructuralReform] | None