Add printing

This commit is contained in:
rstiskalek 2023-10-18 16:30:45 +01:00
parent 960c8b5945
commit 4b038e8d01
2 changed files with 8 additions and 1 deletions

View file

@ -16,7 +16,7 @@ from csiborgtools import clustering, field, match, read, summary
from .utils import (center_of_mass, delta2ncells, number_counts, # noqa
periodic_distance, periodic_distance_two_points, # noqa
binned_statistic, cosine_similarity) # noqa
binned_statistic, cosine_similarity, fprint) # noqa
# Arguments to csiborgtools.read.Paths.

View file

@ -15,6 +15,7 @@
"""Collection of stand-off utility functions used in the scripts."""
import numpy
from numba import jit
from datetime import datetime
###############################################################################
# Positions #
@ -262,3 +263,9 @@ def binned_statistic(x, y, left_edges, bin_width, statistic):
if numpy.any(mask):
out[i] = statistic(y[mask])
return out
def fprint(msg, verbose=True):
"""Print and flush a message with a timestamp."""
if verbose:
print(f"{datetime.now()}: {msg}", flush=True)