diff --git a/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py b/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py index 478c18293c3..2b2edefb7d2 100644 --- a/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py +++ b/sdk/python/feast/infra/online_stores/contrib/cassandra_online_store/cassandra_online_store.py @@ -329,18 +329,22 @@ def __del__(self): you can't use the session object anymore. You'd get a RuntimeError "cannot schedule new futures after shutdown". """ - print( - f"{get_current_time_in_ms()} Calling CassandraOnlineStore __del__() method" - ) + print("Calling CassandraOnlineStore __del__() method") if self._session: if not self._session.is_shutdown: self._session.shutdown() - print(f"{get_current_time_in_ms()} Session is shutdown") + current_time_in_ms = datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[ + :-3 + ] + print(f"{current_time_in_ms} Session is shutdown") if self._cluster: if not self._cluster.is_shutdown: self._cluster.shutdown() - print(f"{get_current_time_in_ms()} Cluster is shutdown") + current_time_in_ms = datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[ + :-3 + ] + print(f"{current_time_in_ms} Cluster is shutdown") def online_write_batch_connector( self,