From cac0bac4265e4fd2aa05406651114c217f1f9a43 Mon Sep 17 00:00:00 2001
From: Drake Eidukas <drake@nominal.io>
Date: Tue, 29 Oct 2024 14:18:22 -0700
Subject: [PATCH] Fix dataset tests

---
 tests/core/test_dataset.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/core/test_dataset.py b/tests/core/test_dataset.py
index 5036027b..782e8701 100644
--- a/tests/core/test_dataset.py
+++ b/tests/core/test_dataset.py
@@ -1,12 +1,12 @@
 import functools
-from datetime import timedelta
+from datetime import datetime, timedelta, timezone
 from unittest.mock import MagicMock, patch
 
 import pytest
 
 from nominal._api.combined.timeseries_logicalseries_api import BatchUpdateLogicalSeriesRequest
 from nominal.core.channel import Channel
-from nominal.core.dataset import Dataset
+from nominal.core.dataset import Dataset, DatasetBounds
 from nominal.core.unit import Unit
 from nominal.exceptions import NominalIngestError, NominalIngestFailed
 
@@ -30,6 +30,10 @@ def mock_dataset(mock_clients):
         rid="test-rid",
         name="Test Dataset",
         description="A dataset for testing",
+        bounds=DatasetBounds(
+            start=datetime(year=2021, month=10, day=9, hour=10, minute=9, second=10, tzinfo=timezone.utc),
+            end=datetime(year=2021, month=10, day=9, hour=11, minute=9, second=10, tzinfo=timezone.utc),
+        ),
         properties={},
         labels=[],
         _clients=mock_clients,