remove prints, add flush to print

Author: LucaMarconato

src/spatialdata_io/readers/cosmx.py

@@ -70,7 +70,6 @@ def cosmx(
     -------
     :class:`spatialdata.SpatialData`
     """
-    print("This is the local version")
     path = Path(path)
 
     # tries to infer dataset_id from the name of the counts file
@@ -107,11 +106,9 @@ def cosmx(
         raise FileNotFoundError(f"Labels directory not found: {labels_dir}.")
 
     counts = pd.read_csv(counts_file, header=0, index_col=CosmxKeys.INSTANCE_KEY)
-    print(counts.head())
     counts.index = counts.index.astype(str).str.cat(counts.pop(CosmxKeys.FOV).astype(str).values, sep="_")
 
     obs = pd.read_csv(meta_file, header=0, index_col=CosmxKeys.INSTANCE_KEY)
-    print(obs.head())
     obs[CosmxKeys.FOV] = pd.Categorical(obs[CosmxKeys.FOV].astype(str))
     obs[CosmxKeys.REGION_KEY] = pd.Categorical(obs[CosmxKeys.FOV].astype(str).apply(lambda s: s + "_labels"))
     obs[CosmxKeys.INSTANCE_KEY] = obs.index.astype(np.int64)
@@ -257,7 +254,7 @@ def cosmx(
         import pyarrow.parquet as pq
 
         with tempfile.TemporaryDirectory() as tmpdir:
-            print("converting .csv to .parquet to improve the speed of the slicing operations... ", end="")
+            print("converting .csv to .parquet to improve the speed of the slicing operations... ", end="", flush=True)
             assert transcripts_file is not None
             transcripts_data = pd.read_csv(transcripts_file, header=0)
             transcripts_data.to_parquet(Path(tmpdir) / "transcripts.parquet")