From e51153f64482d37837209c5c6c6c5f9496729aca Mon Sep 17 00:00:00 2001
From: Shourya Goel <shouryagoel10000@gmail.com>
Date: Thu, 19 Sep 2024 15:25:22 +0530
Subject: [PATCH] [Chore] : Fix Imports (#665)

* Added all_cairo layout and RangeCheck96 tests

* Added a TODO

* nit
---
 pkg/vm/builtins/ecop.go     | 3 +--
 pkg/vm/builtins/pedersen.go | 3 +--
 pkg/vm/builtins/poseidon.go | 3 +--
 3 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/pkg/vm/builtins/ecop.go b/pkg/vm/builtins/ecop.go
index 3f4b0fd8e..592a6c4da 100644
--- 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
index cc9f8d583..df05b57f9 100644
--- 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
index 2d6ed957c..02197e15c 100644
--- 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() {