first use

src/python/pants/engine/internals/native_engine.pyi

@@ -3,7 +3,7 @@
 
 from __future__ import annotations
 
-from datetime import datetime
+from datetime import datetime, timedelta
 from io import RawIOBase
 from typing import (
     Any,
@@ -863,7 +863,7 @@ def session_record_test_observation(
 ) -> None: ...
 def session_isolated_shallow_clone(session: PySession, build_id: str) -> PySession: ...
 def session_wait_for_tail_tasks(
-    scheduler: PyScheduler, session: PySession, timeout: float
+    scheduler: PyScheduler, session: PySession, timeout: timedelta
 ) -> None: ...
 def graph_len(scheduler: PyScheduler) -> int: ...
 def graph_visualize(scheduler: PyScheduler, session: PySession, path: str) -> None: ...

src/python/pants/engine/internals/scheduler.py

@@ -7,6 +7,7 @@
 import os
 import time
 from dataclasses import dataclass
+from datetime import timedelta
 from pathlib import PurePath
 from types import CoroutineType
 from typing import Any, Callable, Dict, Iterable, NoReturn, Sequence, cast
@@ -661,7 +662,7 @@ def is_cancelled(self) -> bool:
     def cancel(self) -> None:
         self.py_session.cancel()
 
-    def wait_for_tail_tasks(self, timeout: float) -> None:
+    def wait_for_tail_tasks(self, timeout: timedelta) -> None:
         native_engine.session_wait_for_tail_tasks(self.py_scheduler, self.py_session, timeout)
 
 

src/rust/engine/src/externs/interface.rs

@@ -35,7 +35,7 @@ use pyo3::prelude::{
     pyclass, pyfunction, pymethods, pymodule, wrap_pyfunction, PyModule, PyObject,
     PyResult as PyO3Result, Python, ToPyObject,
 };
-use pyo3::types::{PyBytes, PyDict, PyList, PyTuple, PyType};
+use pyo3::types::{PyBytes, PyDelta, PyDict, PyList, PyTuple, PyType};
 use pyo3::{create_exception, IntoPy, PyAny, PyRef};
 use regex::Regex;
 use remote::remote_cache::RemoteCacheWarningsBehavior;
@@ -1354,10 +1354,10 @@ fn session_wait_for_tail_tasks(
     py: Python,
     py_scheduler: &PyScheduler,
     py_session: &PySession,
-    timeout: f64,
+    timeout: &PyDelta,
 ) -> PyO3Result<()> {
     let core = &py_scheduler.0.core;
-    let timeout = Duration::from_secs_f64(timeout);
+    let timeout = timeout.extract()?;
     core.executor.enter(|| {
         py.allow_threads(|| {
             core.executor