Remove unused BATCH_DIM, FEATURE_DIM

MilesCranmer · Aug 13, 2023 · 31b2787 · 31b2787
1 parent 14a5670
commit 31b2787
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 9 deletions.
diff --git a/src/Core.jl b/src/Core.jl
@@ -7,8 +7,7 @@ include("OptionsStruct.jl")
 include("Operators.jl")
 include("Options.jl")
 
-import .ProgramConstantsModule:
-    MAX_DEGREE, BATCH_DIM, FEATURE_DIM, RecordType, DATA_TYPE, LOSS_TYPE
+import .ProgramConstantsModule: MAX_DEGREE, RecordType, DATA_TYPE, LOSS_TYPE
 import .DatasetModule: Dataset
 import .OptionsStructModule: Options, ComplexityMapping, MutationWeights, sample_mutation
 import .OptionsModule: Options

diff --git a/src/Dataset.jl b/src/Dataset.jl
@@ -11,7 +11,7 @@ import DynamicQuantities:
 import ...InterfaceDynamicQuantitiesModule: get_si_units, get_sym_units
 
 import ..UtilsModule: subscriptify
-import ..ProgramConstantsModule: BATCH_DIM, FEATURE_DIM, DATA_TYPE, LOSS_TYPE
+import ..ProgramConstantsModule: DATA_TYPE, LOSS_TYPE
 #! format: off
 import ...deprecate_varmap
 #! format: on
@@ -120,8 +120,8 @@ function Dataset(
     # Deprecation warning:
     variable_names = deprecate_varmap(variable_names, varMap, :Dataset)
 
-    n = size(X, BATCH_DIM)
-    nfeatures = size(X, FEATURE_DIM)
+    n = size(X, 2)
+    nfeatures = size(X, 1)
     weighted = weights !== nothing
     variable_names = if variable_names === nothing
         ["x$(i)" for i in 1:nfeatures]

diff --git a/src/ProgramConstants.jl b/src/ProgramConstants.jl
@@ -1,8 +1,6 @@
 module ProgramConstantsModule
 
 const MAX_DEGREE = 2
-const BATCH_DIM = 2
-const FEATURE_DIM = 1
 const RecordType = Dict{String,Any}
 
 const DATA_TYPE = Number

diff --git a/src/SymbolicRegression.jl b/src/SymbolicRegression.jl
@@ -161,8 +161,6 @@ include("SearchUtils.jl")
 
 import .CoreModule:
     MAX_DEGREE,
-    BATCH_DIM,
-    FEATURE_DIM,
     DATA_TYPE,
     LOSS_TYPE,
     RecordType,