@@ -19,7 +19,7 @@ import (
1919 "github.com/Layr-Labs/eigenlayer-cli/pkg/utils"
2020
2121 "github.com/Layr-Labs/eigensdk-go/chainio/clients/elcontracts"
22- contractIAllocationManager "github.com/Layr-Labs/eigensdk-go/contracts/bindings/IAllocationManager "
22+ allocationmanager "github.com/Layr-Labs/eigensdk-go/contracts/bindings/AllocationManager "
2323 "github.com/Layr-Labs/eigensdk-go/logging"
2424 eigenSdkUtils "github.com/Layr-Labs/eigensdk-go/utils"
2525
@@ -200,7 +200,7 @@ func generateAllocationsParams(
200200 config * updateConfig ,
201201 logger logging.Logger ,
202202) (* BulkModifyAllocations , error ) {
203- allocations := make ([]contractIAllocationManager .IAllocationManagerTypesMagnitudeAllocation , 0 )
203+ allocations := make ([]allocationmanager .IAllocationManagerTypesMagnitudeAllocation , 0 )
204204 var allocatableMagnitudes map [gethcommon.Address ]uint64
205205
206206 var err error
@@ -226,10 +226,10 @@ func generateAllocationsParams(
226226 logger .Debugf ("Bips to allocate: %d" , config .bipsToAllocate )
227227 magnitudeToUpdate := calculateMagnitudeToUpdate (magnitude [0 ], config .bipsToAllocate )
228228 logger .Debugf ("Magnitude to update: %d" , magnitudeToUpdate )
229- malloc := contractIAllocationManager .IAllocationManagerTypesMagnitudeAllocation {
229+ malloc := allocationmanager .IAllocationManagerTypesMagnitudeAllocation {
230230 Strategy : config .strategyAddress ,
231231 ExpectedMaxMagnitude : magnitude [0 ],
232- OperatorSets : []contractIAllocationManager .OperatorSet {
232+ OperatorSets : []allocationmanager .OperatorSet {
233233 {
234234 Avs : config .avsAddress ,
235235 OperatorSetId : config .operatorSetId ,
@@ -255,7 +255,7 @@ func computeAllocations(
255255 filePath string ,
256256 operatorAddress gethcommon.Address ,
257257 elReader elChainReader ,
258- ) ([]contractIAllocationManager .IAllocationManagerTypesMagnitudeAllocation , map [gethcommon.Address ]uint64 , error ) {
258+ ) ([]allocationmanager .IAllocationManagerTypesMagnitudeAllocation , map [gethcommon.Address ]uint64 , error ) {
259259 allocations , err := parseAllocationsCSV (filePath )
260260 if err != nil {
261261 return nil , nil , eigenSdkUtils .WrapError ("failed to parse allocations csv" , err )
@@ -385,19 +385,19 @@ func parseAllocationsCSV(filePath string) ([]allocation, error) {
385385func convertAllocationsToMagnitudeAllocations (
386386 allocations []allocation ,
387387 strategyTotalMagnitudes map [gethcommon.Address ]uint64 ,
388- ) []contractIAllocationManager .IAllocationManagerTypesMagnitudeAllocation {
389- magnitudeAllocations := make ([]contractIAllocationManager .IAllocationManagerTypesMagnitudeAllocation , 0 )
390- operatorSetsPerStragyMap := make (map [gethcommon.Address ][]contractIAllocationManager .OperatorSet )
388+ ) []allocationmanager .IAllocationManagerTypesMagnitudeAllocation {
389+ magnitudeAllocations := make ([]allocationmanager .IAllocationManagerTypesMagnitudeAllocation , 0 )
390+ operatorSetsPerStragyMap := make (map [gethcommon.Address ][]allocationmanager .OperatorSet )
391391 magnitudeAllocationsPerStrategyMap := make (map [gethcommon.Address ][]uint64 )
392392 for _ , a := range allocations {
393393 totalMag := strategyTotalMagnitudes [a .StrategyAddress ]
394394 magnitudeToUpdate := calculateMagnitudeToUpdate (totalMag , a .Bips )
395395
396396 operatorSets , ok := operatorSetsPerStragyMap [a .StrategyAddress ]
397397 if ! ok {
398- operatorSets = make ([]contractIAllocationManager .OperatorSet , 0 )
398+ operatorSets = make ([]allocationmanager .OperatorSet , 0 )
399399 }
400- operatorSets = append (operatorSets , contractIAllocationManager .OperatorSet {
400+ operatorSets = append (operatorSets , allocationmanager .OperatorSet {
401401 Avs : a .AvsAddress ,
402402 OperatorSetId : a .OperatorSetId ,
403403 })
@@ -411,7 +411,7 @@ func convertAllocationsToMagnitudeAllocations(
411411 for strategy , operatorSets := range operatorSetsPerStragyMap {
412412 magnitudeAllocations = append (
413413 magnitudeAllocations ,
414- contractIAllocationManager .IAllocationManagerTypesMagnitudeAllocation {
414+ allocationmanager .IAllocationManagerTypesMagnitudeAllocation {
415415 Strategy : strategy ,
416416 ExpectedMaxMagnitude : strategyTotalMagnitudes [strategy ],
417417 OperatorSets : operatorSets ,
0 commit comments