Skip to content

Commit 8338529

Browse files
authored
Avoid using timezone-aware datetime data in GPU tables, unpin uvicorn in 3.8 GPU environments (#1116)
* Avoid using timezoned data in GPU datetime tables * Unpin uvicorn dependency in GPU 3.8 environment
1 parent e107382 commit 8338529

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

continuous_integration/gpuci/environment-3.8.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ dependencies:
3737
- sqlalchemy<2
3838
- tpot
3939
- tzlocal=2.1
40-
- uvicorn=0.13.4
40+
- uvicorn>=0.13.4
4141
# GPU-specific requirements
4242
- cudatoolkit=11.5
4343
- cudf=23.06

tests/integration/fixtures.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,11 @@ def gpu_string_table(string_table):
163163

164164
@pytest.fixture()
165165
def gpu_datetime_table(datetime_table):
166+
# cudf doesn't have support for timezoned datetime data
167+
datetime_table["timezone"] = datetime_table["timezone"].astype("datetime64[ns]")
168+
datetime_table["utc_timezone"] = datetime_table["utc_timezone"].astype(
169+
"datetime64[ns]"
170+
)
166171
return cudf.from_pandas(datetime_table) if cudf else None
167172

168173

0 commit comments

Comments
 (0)