Move error types to types modules

packages/squiggle-lang/__tests__/ReducerInterface/ReducerInterface_Distribution_test.res

@@ -88,10 +88,7 @@ describe("eval on distribution functions", () => {
 
   describe("log", () => {
     testEval("log(2, uniform(5,8))", "Ok(Sample Set Distribution)")
-    testEval(
+    testEval("log(normal(5,2), 3)", "Error(Math Error: Operation returned complex result)")
-      "log(normal(5,2), 3)",
-      "Error(Math Error: Operation returned complex result)",
-    )
     testEval(
       "log(normal(5,2), normal(10,1))",
       "Error(Math Error: Operation returned complex result)",

packages/squiggle-lang/src/rescript/Distributions/DistributionTypes.res

@@ -11,16 +11,11 @@ type error =
   | DistributionVerticalShiftIsInvalid
   | TooFewSamples
   | ArgumentError(string)
-  | OperationError(Operation.invalidOperationError)
+  | OperationError(Operation.Error.invalidOperationError)
   | PointSetConversionError(SampleSetDist.pointsetConversionError)
-  | SparklineError(PointSetDist.sparklineError) // This type of error is for when we find a sparkline of a discrete distribution. This should probably at some point be actually implemented
+  | SparklineError(PointSetTypes.sparklineError) // This type of error is for when we find a sparkline of a discrete distribution. This should probably at some point be actually implemented
   | Other(string)
 
-let sampleErrorToDistErr = (err: SampleSetDist.sampleSetError): error =>
-  switch err {
-  | TooFewSamples => TooFewSamples
-  }
-
 @genType
 module Error = {
   type t = error
@@ -35,14 +30,19 @@ module Error = {
     | DistributionVerticalShiftIsInvalid => "Distribution Vertical Shift is Invalid"
     | ArgumentError(s) => `Argument Error ${s}`
     | TooFewSamples => "Too Few Samples"
-    | OperationError(err) => Operation.invalidOperationErrorToString(err)
+    | OperationError(err) => Operation.Error.invalidOperationErrorToString(err)
     | PointSetConversionError(err) => SampleSetDist.pointsetConversionErrorToString(err)
-    | SparklineError(err) => PointSetDist.sparklineErrorToString(err)
+    | SparklineError(err) => PointSetTypes.sparklineErrorToString(err)
     | Other(s) => s
     }
 
   let resultStringToResultError: result<'a, string> => result<'a, error> = n =>
     n->E.R2.errMap(r => r->fromString)
+
+  let sampleErrorToDistErr = (err: SampleSetDist.sampleSetError): error =>
+    switch err {
+    | TooFewSamples => TooFewSamples
+    }
 }
 
 @genType

packages/squiggle-lang/src/rescript/Distributions/GenericDist/GenericDist.res

@@ -14,7 +14,7 @@ let sampleN = (t: t, n) =>
   }
 
 let toSampleSetDist = (t: t, n) =>
-  SampleSetDist.make(sampleN(t, n))->E.R2.errMap(DistributionTypes.sampleErrorToDistErr)
+  SampleSetDist.make(sampleN(t, n))->E.R2.errMap(DistributionTypes.Error.sampleErrorToDistErr)
 
 let fromFloat = (f: float): t => Symbolic(SymbolicDist.Float.make(f))
 
@@ -151,7 +151,7 @@ module AlgebraicCombination = {
     arithmeticOperation: DistributionTypes.Operation.arithmeticOperation,
     t1: t,
     t2: t,
-  ): option<result<SymbolicDistTypes.symbolicDist, Operation.invalidOperationError>> =>
+  ): option<result<SymbolicDistTypes.symbolicDist, Operation.Error.invalidOperationError>> =>
     switch (arithmeticOperation, t1, t2) {
     | (arithmeticOperation, Symbolic(d1), Symbolic(d2)) =>
       switch SymbolicDist.T.tryAnalyticalSimplification(d1, d2, arithmeticOperation) {

packages/squiggle-lang/src/rescript/Distributions/PointSetDist/PointSetDist.res

@@ -215,16 +215,8 @@ let operate = (distToFloatOp: Operation.distToFloatOperation, s): float =>
   | #Mean => T.mean(s)
   }
 
-@genType
+let toSparkline = (t: t, bucketCount): result<string, PointSetTypes.sparklineError> =>
-type sparklineError = CannotSparklineDiscrete
-
-let sparklineErrorToString = (err: sparklineError): string =>
-  switch err {
-  | CannotSparklineDiscrete => "Cannot find the sparkline of a discrete distribution"
-  }
-
-let toSparkline = (t: t, bucketCount): result<string, sparklineError> =>
   T.toContinuous(t)
   ->E.O2.fmap(Continuous.downsampleEquallyOverX(bucketCount))
-  ->E.O2.toResult(CannotSparklineDiscrete)
+  ->E.O2.toResult(PointSetTypes.CannotSparklineDiscrete)
   ->E.R2.fmap(r => Continuous.getShape(r).ys->Sparklines.create())

packages/squiggle-lang/src/rescript/Distributions/PointSetDist/PointSetTypes.res

@@ -94,3 +94,11 @@ module MixedPoint = {
 
   let add = combine2((a, b) => a +. b)
 }
+
+@genType
+type sparklineError = CannotSparklineDiscrete
+
+let sparklineErrorToString = (err: sparklineError): string =>
+  switch err {
+  | CannotSparklineDiscrete => "Cannot find the sparkline of a discrete distribution"
+  }

packages/squiggle-lang/src/rescript/Distributions/SampleSetDist/SampleSetDist.res

@@ -1,10 +1,23 @@
 @genType
-type sampleSetError = TooFewSamples
+module Error = {
+  @genType
+  type sampleSetError = TooFewSamples
 
-let sampleSetErrorToString = (err: sampleSetError): string =>
+  let sampleSetErrorToString = (err: sampleSetError): string =>
-  switch err {
+    switch err {
-  | TooFewSamples => "Too few samples when constructing sample set"
+    | TooFewSamples => "Too few samples when constructing sample set"
-  }
+    }
+
+  @genType
+  type pointsetConversionError = TooFewSamplesForConversionToPointSet
+
+  let pointsetConversionErrorToString = (err: pointsetConversionError) =>
+    switch err {
+    | TooFewSamplesForConversionToPointSet => "Too Few Samples to convert to point set"
+    }
+}
+
+include Error
 
 /*
 This is used as a smart constructor. The only way to create a SampleSetDist.t is to call
@@ -33,14 +46,6 @@ include T
 
 let length = (t: t) => get(t)->E.A.length
 
-@genType
-type pointsetConversionError = TooFewSamplesForConversionToPointSet
-
-let pointsetConversionErrorToString = (err: pointsetConversionError) =>
-  switch err {
-  | TooFewSamplesForConversionToPointSet => "Too Few Samples to convert to point set"
-  }
-
 /*
 TODO: Refactor to get a more precise estimate. Also, this code is just fairly messy, could use 
 some refactoring.
@@ -79,10 +84,10 @@ let sampleN = (t: t, n) => {
 
 //TODO: Figure out what to do if distributions are different lengths. ``zip`` is kind of inelegant for this.
 let map2 = (
-  ~fn: (float, float) => result<float, Operation.invalidOperationError>,
+  ~fn: (float, float) => result<float, Operation.Error.invalidOperationError>,
   ~t1: t,
   ~t2: t,
-): result<t, Operation.invalidOperationError> => {
+): result<t, Operation.Error.invalidOperationError> => {
   let samples = Belt.Array.zip(get(t1), get(t2))->E.A2.fmap(((a, b)) => fn(a, b))
 
   // This assertion should never be reached. In order for it to be reached, one

packages/squiggle-lang/src/rescript/Distributions/SymbolicDist/SymbolicDistTypes.res

@@ -45,6 +45,6 @@ type symbolicDist = [
 
 type analyticalSimplificationResult = [
   | #AnalyticalSolution(symbolicDist)
-  | #Error(Operation.invalidOperationError)
+  | #Error(Operation.Error.invalidOperationError)
   | #NoSolution
 ]

packages/squiggle-lang/src/rescript/Utility/Operation.res

@@ -38,19 +38,22 @@ module Convolution = {
 }
 
 @genType
-type invalidOperationError =
+module Error = {
-  | DivisionByZeroError
+  @genType
-  | ComplexNumberError
+  type invalidOperationError =
+    | DivisionByZeroError
+    | ComplexNumberError
 
-let invalidOperationErrorToString = (err: invalidOperationError): string =>
+  let invalidOperationErrorToString = (err: invalidOperationError): string =>
-  switch err {
+    switch err {
-  | DivisionByZeroError => "Cannot divide by zero"
+    | DivisionByZeroError => "Cannot divide by zero"
-  | ComplexNumberError => "Operation returned complex result"
+    | ComplexNumberError => "Operation returned complex result"
-  }
+    }
+}
 
 module Algebraic = {
   type t = algebraicOperation
-  let toFn: (t, float, float) => result<float, invalidOperationError> = (x, a, b) =>
+  let toFn: (t, float, float) => result<float, Error.invalidOperationError> = (x, a, b) =>
     switch x {
     | #Add => Ok(a +. b)
     | #Subtract => Ok(a -. b)
@@ -70,8 +73,7 @@ module Algebraic = {
     | #Logarithm =>
       if b == 1. {
         Error(DivisionByZeroError)
-      }
+      } else if a > 0.0 && b > 0.0 {
-      else if a > 0.0 && b > 0.0 {
         Ok(log(a) /. log(b))
       } else {
         Error(ComplexNumberError)
@@ -119,7 +121,7 @@ module DistToFloat = {
 // Note that different logarithms don't really do anything.
 module Scale = {
   type t = scaleOperation
-  let toFn = (x: t, a: float, b: float): result<float, invalidOperationError> =>
+  let toFn = (x: t, a: float, b: float): result<float, Error.invalidOperationError> =>
     switch x {
     | #Multiply => Ok(a *. b)
     | #Divide =>