Merge pull request PAIR-code#251 from tvalentyn/asscalar

Remove remaining instances of asscalar.
kp-forks · Feb 6, 2023 · 164085a · 164085a
2 parents a56bc97 + 0a492bb
commit 164085a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/facets_overview/facets_overview/base_generic_feature_statistics_generator.py b/facets_overview/facets_overview/base_generic_feature_statistics_generator.py
@@ -232,7 +232,7 @@ def GetDatasetsProto(self, datasets, features=None,
                 bucket = hist.buckets.add(
                     low_value=buckets[bucket_count],
                     high_value=buckets[bucket_count + 1],
-                    sample_count=np.asscalar(counts[bucket_count]))
+                    sample_count=counts[bucket_count].item())
                 # Add any negative or positive infinities to the first and last
                 # buckets in the histogram.
                 if bucket_count == 0 and num_neginf > 0:
@@ -281,7 +281,7 @@ def GetDatasetsProto(self, datasets, features=None,
                 bucket = featstats.rank_histogram.buckets.add(
                     low_rank=val_index,
                     high_rank=val_index,
-                    sample_count=np.asscalar(val[0]),
+                    sample_count=val[0].item(),
                     label=printable_val)
                 if val_index < 2:
                   featstats.top_values.add(

diff --git a/facets_overview/python/base_generic_feature_statistics_generator.py b/facets_overview/python/base_generic_feature_statistics_generator.py
@@ -237,7 +237,7 @@ def GetDatasetsProto(self, datasets, features=None,
                 bucket = hist.buckets.add(
                     low_value=buckets[bucket_count],
                     high_value=buckets[bucket_count + 1],
-                    sample_count=np.asscalar(counts[bucket_count]))
+                    sample_count=counts[bucket_count].item())
                 # Add any negative or positive infinities to the first and last
                 # buckets in the histogram.
                 if bucket_count == 0 and num_neginf > 0:
@@ -286,7 +286,7 @@ def GetDatasetsProto(self, datasets, features=None,
                 bucket = featstats.rank_histogram.buckets.add(
                     low_rank=val_index,
                     high_rank=val_index,
-                    sample_count=np.asscalar(val[0]),
+                    sample_count=val[0].item(),
                     label=printable_val)
                 if val_index < 2:
                   featstats.top_values.add(