Merge pull request #233 from MilesCranmer/allow-empty-units

Allow empty string units (==1)

Author: MilesCranmer

Project.toml

@@ -1,7 +1,7 @@
 name = "SymbolicRegression"
 uuid = "8254be44-1295-4e6a-a16d-46603ac705cb"
 authors = ["MilesCranmer <miles.cranmer@gmail.com>"]
-version = "0.21.1"
+version = "0.21.2"
 
 [deps]
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"

src/Dataset.jl

@@ -224,6 +224,7 @@ function get_units(_, _, ::Nothing, ::Function)
     return nothing
 end
 function get_units(::Type{T}, ::Type{D}, x::AbstractString, f::Function) where {T,D}
+    isempty(x) && return one(Quantity{T,D})
     return convert(Quantity{T,D}, f(x))
 end
 function get_units(::Type{T}, ::Type{D}, x::Quantity, ::Function) where {T,D}

test/test_units.jl

@@ -44,6 +44,7 @@ options = Options(;
         Quantity(1.5, SymbolicDimensions),
         Quantity(1.0, SymbolicDimensions),
     ]
+    @test get_units(Float64, SD, [""], sym_uparse) == [Quantity(1.0, SymbolicDimensions)]
     # Bad unit types:
     @test_throws ErrorException get_units(Float64, D, (; X=[1, 2]), uparse)