symbolicai 0.20.2__py3-none-any.whl → 1.0.0__py3-none-any.whl

symai/ops/__init__.py

@@ -1,25 +1,35 @@
-from .primitives import *
+from . import primitives as _primitives
+
+__all__ = getattr(_primitives, "__all__", None) # noqa
+if __all__ is None:
+    __all__ = [name for name in dir(_primitives) if not name.startswith("_")]
+
+for _name in __all__:
+    globals()[_name] = getattr(_primitives, _name)
 
 SYMBOL_PRIMITIVES = [
-    OperatorPrimitives,
-    IterationPrimitives,
-    ValueHandlingPrimitives,
-    StringHelperPrimitives,
-    CastingPrimitives,
-    ComparisonPrimitives,
-    ExpressionHandlingPrimitives,
-    DataHandlingPrimitives,
-    UniquenessPrimitives,
-    PatternMatchingPrimitives,
-    DictHandlingPrimitives,
-    QueryHandlingPrimitives,
-    ExecutionControlPrimitives,
-    TemplateStylingPrimitives,
-    DataClusteringPrimitives,
-    EmbeddingPrimitives,
-    IndexingPrimitives,
-    IOHandlingPrimitives,
-    PersistencePrimitives,
-    OutputHandlingPrimitives,
-    FineTuningPrimitives,
+    _primitives.OperatorPrimitives,
+    _primitives.IterationPrimitives,
+    _primitives.ValueHandlingPrimitives,
+    _primitives.StringHelperPrimitives,
+    _primitives.CastingPrimitives,
+    _primitives.ComparisonPrimitives,
+    _primitives.ExpressionHandlingPrimitives,
+    _primitives.DataHandlingPrimitives,
+    _primitives.UniquenessPrimitives,
+    _primitives.PatternMatchingPrimitives,
+    _primitives.DictHandlingPrimitives,
+    _primitives.QueryHandlingPrimitives,
+    _primitives.ExecutionControlPrimitives,
+    _primitives.TemplateStylingPrimitives,
+    _primitives.DataClusteringPrimitives,
+    _primitives.EmbeddingPrimitives,
+    _primitives.IndexingPrimitives,
+    _primitives.IOHandlingPrimitives,
+    _primitives.PersistencePrimitives,
+    _primitives.OutputHandlingPrimitives,
+    _primitives.FineTuningPrimitives,
 ]
+
+del _name
+del _primitives

symai/ops/measures.py

@@ -1,7 +1,8 @@
 import numpy as np
-
 from scipy import linalg
 
+from ..utils import UserMessage
+
 
 def calculate_frechet_distance(mu1, sigma1, mu2, sigma2, eps=1e-6):
     """Numpy implementation of the Frechet Distance.
@@ -42,8 +43,8 @@ def calculate_frechet_distance(mu1, sigma1, mu2, sigma2, eps=1e-6):
     covmean, _ = linalg.sqrtm(sigma1.dot(sigma2), disp=False)
     if not np.isfinite(covmean).all():
         msg = ('fid calculation produces singular product; '
-            'adding %s to diagonal of cov estimates') % eps
-        print(msg)
+            f'adding {eps} to diagonal of cov estimates')
+        UserMessage(msg)
         offset = np.eye(sigma1.shape[0]) * eps
         covmean = linalg.sqrtm((sigma1 + offset).dot(sigma2 + offset))
 
@@ -51,12 +52,11 @@ def calculate_frechet_distance(mu1, sigma1, mu2, sigma2, eps=1e-6):
     if np.iscomplexobj(covmean):
         if not np.allclose(np.diagonal(covmean).imag, 0, atol=1e-3):
             m = np.max(np.abs(covmean.imag))
-            raise ValueError('Imaginary component {}'.format(m))
+            UserMessage(f'Imaginary component {m}', raise_with=ValueError)
         covmean = covmean.real
 
     tr_covmean = np.trace(covmean)
-    val = diff.dot(diff) + np.trace(sigma1) + np.trace(sigma2) - 2 * tr_covmean
-    return val
+    return diff.dot(diff) + np.trace(sigma1) + np.trace(sigma2) - 2 * tr_covmean
 
 
 def calculate_mmd(x, y, kernel='rbf', kernel_mul=2.0, kernel_num=5, fix_sigma=None, eps=1e-9):
@@ -67,10 +67,7 @@ def calculate_mmd(x, y, kernel='rbf', kernel_mul=2.0, kernel_num=5, fix_sigma=No
         total1 = np.expand_dims(total, 1)
         L2_distance = np.sum((total0 - total1) ** 2, axis=2)
 
-        if fix_sigma:
-            bandwidth = fix_sigma
-        else:
-            bandwidth = np.sum(L2_distance) / (n_samples ** 2 - n_samples + eps)
+        bandwidth = fix_sigma or np.sum(L2_distance) / (n_samples ** 2 - n_samples + eps)
         bandwidth /= kernel_mul ** (kernel_num // 2)
         bandwidth_list = [bandwidth * (kernel_mul ** i) for i in range(kernel_num)]
         kernel_val = [np.exp(-L2_distance / (bandwidth_temp + eps)) for bandwidth_temp in bandwidth_list]
@@ -78,17 +75,16 @@ def calculate_mmd(x, y, kernel='rbf', kernel_mul=2.0, kernel_num=5, fix_sigma=No
 
     def linear_mmd2(f_of_X, f_of_Y):
         delta = f_of_X.mean(axis=0) - f_of_Y.mean(axis=0)
-        loss = np.dot(delta, delta.T)
-        return loss
+        return np.dot(delta, delta.T)
 
     if kernel == 'linear':
         return linear_mmd2(x, y)
-    elif kernel == 'rbf':
+    if kernel == 'rbf':
         batch_size = x.shape[0]
         kernels = gaussian_kernel(x, y, kernel_mul=kernel_mul, kernel_num=kernel_num, fix_sigma=fix_sigma)
         xx = np.mean(kernels[:batch_size, :batch_size])
         yy = np.mean(kernels[batch_size:, batch_size:])
         xy = np.mean(kernels[:batch_size, batch_size:])
         yx = np.mean(kernels[batch_size:, :batch_size])
-        loss = xx + yy - xy - yx
-        return loss
+        return xx + yy - xy - yx
+    return None