[Chore] : Fix Imports (#665)

* Added all_cairo layout and RangeCheck96 tests * Added a TODO * nit
NethermindEth · Sep 19, 2024 · e51153f · e51153f
1 parent 5f40559
commit e51153f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/pkg/vm/builtins/ecop.go b/pkg/vm/builtins/ecop.go
@@ -7,7 +7,6 @@ import (
 	"sort"
 
 	"github.com/NethermindEth/cairo-vm-go/pkg/utils"
-	"github.com/NethermindEth/cairo-vm-go/pkg/vm/memory"
 	mem "github.com/NethermindEth/cairo-vm-go/pkg/vm/memory"
 	"github.com/consensys/gnark-crypto/ecc/stark-curve/fp"
 	"github.com/holiman/uint256"
@@ -247,7 +246,7 @@ type AirPrivateBuiltinEcOp struct {
 	QY    string `json:"q_y"`
 }
 
-func (e *EcOp) GetAirPrivateInput(ecOpSegment *memory.Segment) []AirPrivateBuiltinEcOp {
+func (e *EcOp) GetAirPrivateInput(ecOpSegment *mem.Segment) []AirPrivateBuiltinEcOp {
 	valueMapping := make(map[int]AirPrivateBuiltinEcOp)
 	for index, value := range ecOpSegment.Data {
 		if !value.Known() {

diff --git a/pkg/vm/builtins/pedersen.go b/pkg/vm/builtins/pedersen.go
@@ -6,7 +6,6 @@ import (
 	"math/big"
 	"sort"
 
-	"github.com/NethermindEth/cairo-vm-go/pkg/vm/memory"
 	mem "github.com/NethermindEth/cairo-vm-go/pkg/vm/memory"
 	pedersenhash "github.com/consensys/gnark-crypto/ecc/stark-curve/pedersen-hash"
 )
@@ -73,7 +72,7 @@ type AirPrivateBuiltinPedersen struct {
 	Y     string `json:"y"`
 }
 
-func (p *Pedersen) GetAirPrivateInput(pedersenSegment *memory.Segment) []AirPrivateBuiltinPedersen {
+func (p *Pedersen) GetAirPrivateInput(pedersenSegment *mem.Segment) []AirPrivateBuiltinPedersen {
 	valueMapping := make(map[int]AirPrivateBuiltinPedersen)
 	for index, value := range pedersenSegment.Data {
 		if !value.Known() {

diff --git a/pkg/vm/builtins/poseidon.go b/pkg/vm/builtins/poseidon.go
@@ -6,7 +6,6 @@ import (
 	"math/big"
 	"sort"
 
-	"github.com/NethermindEth/cairo-vm-go/pkg/vm/memory"
 	mem "github.com/NethermindEth/cairo-vm-go/pkg/vm/memory"
 	"github.com/consensys/gnark-crypto/ecc/stark-curve/fp"
 )
@@ -74,7 +73,7 @@ type AirPrivateBuiltinPoseidon struct {
 	InputS2 string `json:"input_s2"`
 }
 
-func (p *Poseidon) GetAirPrivateInput(poseidonSegment *memory.Segment) []AirPrivateBuiltinPoseidon {
+func (p *Poseidon) GetAirPrivateInput(poseidonSegment *mem.Segment) []AirPrivateBuiltinPoseidon {
 	valueMapping := make(map[int]AirPrivateBuiltinPoseidon)
 	for index, value := range poseidonSegment.Data {
 		if !value.Known() {