@@ -15,10 +15,6 @@ class Database:
1515
1616 :param array_like parameters: the input parameters
1717 :param array_like snapshots: the input snapshots
18- :param Scale scaler_parameters: the scaler for the parameters. Default
19- is None meaning no scaling.
20- :param Scale scaler_snapshots: the scaler for the snapshots. Default is
21- None meaning no scaling.
2218 :param array_like space: the input spatial data
2319
2420 :Example:
@@ -46,6 +42,7 @@ def __init__(self, parameters=None, snapshots=None, space=None):
4642 )
4743 self ._pairs = []
4844
45+
4946 if parameters is None and snapshots is None :
5047 logger .debug ("Empty database created" )
5148 return
@@ -149,19 +146,19 @@ def add(self, parameter, snapshot):
149146 """
150147 if not isinstance (parameter , Parameter ):
151148 logger .error ("Invalid parameter type: %s" , type (parameter ))
152- raise ValueError
149+ raise TypeError ( f"Expected a Parameter object, got { type ( parameter ) } " )
153150
154151 if not isinstance (snapshot , Snapshot ):
155152 logger .error ("Invalid snapshot type: %s" , type (snapshot ))
156- raise ValueError
153+ raise TypeError ( f"Expected a Snapshot object, got { type ( snapshot ) } " )
157154
158155 self ._pairs .append ((parameter , snapshot ))
159156 logger .debug (
160157 "Added parameter-snapshot pair. Total pairs: %d" , len (self ._pairs )
161158 )
162159
163160 return self
164-
161+
165162 def split (self , chunks , seed = None ):
166163 """
167164
@@ -209,7 +206,7 @@ def split(self, chunks, seed=None):
209206
210207 else :
211208 logger .error ("Invalid chunk type" )
212- ValueError
209+ raise TypeError ( f"Invalid chunk type. Expected a list of integers or floats, but got { type ( chunks ) } ." )
213210
214211 new_database = [Database () for _ in range (len (chunks ))]
215212 for i , chunk in enumerate (chunks ):
@@ -235,4 +232,4 @@ def get_snapshot_space(self, index):
235232 """
236233 if index < 0 or index >= len (self ._pairs ):
237234 raise IndexError ("Snapshot index out of range." )
238- return self ._pairs [index ][1 ].space
235+ return self ._pairs [index ][1 ].space
0 commit comments