diff --git a/cmd/spark-connect-example-raw-grpc-client/main.go b/cmd/spark-connect-example-raw-grpc-client/main.go index 1f463db..ab3c405 100644 --- a/cmd/spark-connect-example-raw-grpc-client/main.go +++ b/cmd/spark-connect-example-raw-grpc-client/main.go @@ -22,7 +22,7 @@ import ( "log" "time" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" "github.com/google/uuid" "google.golang.org/grpc" "google.golang.org/grpc/credentials/insecure" diff --git a/cmd/spark-connect-example-spark-session/main.go b/cmd/spark-connect-example-spark-session/main.go index ec720dc..a05758b 100644 --- a/cmd/spark-connect-example-spark-session/main.go +++ b/cmd/spark-connect-example-spark-session/main.go @@ -22,12 +22,12 @@ import ( "fmt" "log" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - "github.com/apache/spark-connect-go/spark/sql/functions" + "github.com/caldempsey/spark-connect-go/spark/sql/functions" - "github.com/apache/spark-connect-go/spark/sql" - "github.com/apache/spark-connect-go/spark/sql/utils" + "github.com/caldempsey/spark-connect-go/spark/sql" + "github.com/caldempsey/spark-connect-go/spark/sql/utils" ) var ( diff --git a/go.mod b/go.mod index 3478ddb..379f88e 100644 --- a/go.mod +++ b/go.mod @@ -13,7 +13,7 @@ // See the License for the specific language governing permissions and // limitations under the License. -module github.com/apache/spark-connect-go +module github.com/caldempsey/spark-connect-go go 1.23.2 diff --git a/internal/tests/integration/dataframe_test.go b/internal/tests/integration/dataframe_test.go index d383ca1..5978765 100644 --- a/internal/tests/integration/dataframe_test.go +++ b/internal/tests/integration/dataframe_test.go @@ -21,15 +21,15 @@ import ( "os" "testing" - "github.com/apache/spark-connect-go/spark/sql/utils" + "github.com/caldempsey/spark-connect-go/spark/sql/utils" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - "github.com/apache/spark-connect-go/spark/sql/column" + "github.com/caldempsey/spark-connect-go/spark/sql/column" - "github.com/apache/spark-connect-go/spark/sql/functions" + "github.com/caldempsey/spark-connect-go/spark/sql/functions" - "github.com/apache/spark-connect-go/spark/sql" + "github.com/caldempsey/spark-connect-go/spark/sql" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) diff --git a/internal/tests/integration/functions_test.go b/internal/tests/integration/functions_test.go index 94310d7..18c8cb2 100644 --- a/internal/tests/integration/functions_test.go +++ b/internal/tests/integration/functions_test.go @@ -19,11 +19,11 @@ import ( "context" "testing" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - "github.com/apache/spark-connect-go/spark/sql/functions" + "github.com/caldempsey/spark-connect-go/spark/sql/functions" - "github.com/apache/spark-connect-go/spark/sql" + "github.com/caldempsey/spark-connect-go/spark/sql" "github.com/stretchr/testify/assert" ) diff --git a/internal/tests/integration/helper.go b/internal/tests/integration/helper.go index 902d223..df0b630 100644 --- a/internal/tests/integration/helper.go +++ b/internal/tests/integration/helper.go @@ -22,7 +22,7 @@ import ( "github.com/apache/arrow-go/v18/arrow/array" "github.com/apache/arrow-go/v18/arrow/memory" - "github.com/apache/spark-connect-go/spark/sql" + "github.com/caldempsey/spark-connect-go/spark/sql" ) func connect() (context.Context, sql.SparkSession) { diff --git a/internal/tests/integration/spark_runner.go b/internal/tests/integration/spark_runner.go index b6cb688..5796456 100644 --- a/internal/tests/integration/spark_runner.go +++ b/internal/tests/integration/spark_runner.go @@ -23,7 +23,7 @@ import ( "os/exec" "time" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) func StartSparkConnect() (int64, error) { diff --git a/internal/tests/integration/sql_test.go b/internal/tests/integration/sql_test.go index a0e0493..395b285 100644 --- a/internal/tests/integration/sql_test.go +++ b/internal/tests/integration/sql_test.go @@ -22,13 +22,13 @@ import ( "os" "testing" - "github.com/apache/spark-connect-go/spark/sql/column" + "github.com/caldempsey/spark-connect-go/spark/sql/column" - "github.com/apache/spark-connect-go/spark/sql/functions" + "github.com/caldempsey/spark-connect-go/spark/sql/functions" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - "github.com/apache/spark-connect-go/spark/sql" + "github.com/caldempsey/spark-connect-go/spark/sql" "github.com/stretchr/testify/assert" ) diff --git a/spark/client/base/base.go b/spark/client/base/base.go index d7be261..24e80d2 100644 --- a/spark/client/base/base.go +++ b/spark/client/base/base.go @@ -18,11 +18,11 @@ package base import ( "context" - "github.com/apache/spark-connect-go/spark/sql/utils" + "github.com/caldempsey/spark-connect-go/spark/sql/utils" "github.com/apache/arrow-go/v18/arrow" - "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sql/types" ) type SparkConnectRPCClient generated.SparkConnectServiceClient diff --git a/spark/client/channel/channel.go b/spark/client/channel/channel.go index 6403566..cfca0f3 100644 --- a/spark/client/channel/channel.go +++ b/spark/client/channel/channel.go @@ -29,13 +29,13 @@ import ( "strconv" "strings" - "github.com/apache/spark-connect-go/spark" + "github.com/caldempsey/spark-connect-go/spark" "github.com/google/uuid" "google.golang.org/grpc/credentials/insecure" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" "golang.org/x/oauth2" "google.golang.org/grpc" "google.golang.org/grpc/credentials" diff --git a/spark/client/channel/channel_test.go b/spark/client/channel/channel_test.go index b0f7bea..e4abfc8 100644 --- a/spark/client/channel/channel_test.go +++ b/spark/client/channel/channel_test.go @@ -23,8 +23,8 @@ import ( "github.com/google/uuid" - "github.com/apache/spark-connect-go/spark/client/channel" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/client/channel" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" "github.com/stretchr/testify/assert" ) diff --git a/spark/client/client.go b/spark/client/client.go index 3851292..96bc582 100644 --- a/spark/client/client.go +++ b/spark/client/client.go @@ -21,24 +21,24 @@ import ( "fmt" "io" - "github.com/apache/spark-connect-go/spark/sql/utils" + "github.com/caldempsey/spark-connect-go/spark/sql/utils" "google.golang.org/grpc" "google.golang.org/grpc/metadata" - "github.com/apache/spark-connect-go/spark/client/base" - "github.com/apache/spark-connect-go/spark/mocks" + "github.com/caldempsey/spark-connect-go/spark/client/base" + "github.com/caldempsey/spark-connect-go/spark/mocks" - "github.com/apache/spark-connect-go/spark/client/options" + "github.com/caldempsey/spark-connect-go/spark/client/options" "github.com/google/uuid" "github.com/apache/arrow-go/v18/arrow" "github.com/apache/arrow-go/v18/arrow/array" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) type sparkConnectClientImpl struct { diff --git a/spark/client/client_test.go b/spark/client/client_test.go index aefc14f..ab7524d 100644 --- a/spark/client/client_test.go +++ b/spark/client/client_test.go @@ -23,11 +23,11 @@ import ( "github.com/apache/arrow-go/v18/arrow/array" "github.com/apache/arrow-go/v18/arrow/ipc" "github.com/apache/arrow-go/v18/arrow/memory" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/client" - "github.com/apache/spark-connect-go/spark/client/testutils" - "github.com/apache/spark-connect-go/spark/mocks" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/client" + "github.com/caldempsey/spark-connect-go/spark/client/testutils" + "github.com/caldempsey/spark-connect-go/spark/mocks" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" "github.com/google/uuid" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" diff --git a/spark/client/conf.go b/spark/client/conf.go index 11b301e..22a81e2 100644 --- a/spark/client/conf.go +++ b/spark/client/conf.go @@ -18,8 +18,8 @@ package client import ( "context" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/client/base" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/client/base" ) // Public interface RuntimeConfig diff --git a/spark/client/retry.go b/spark/client/retry.go index be7d90c..016aa46 100644 --- a/spark/client/retry.go +++ b/spark/client/retry.go @@ -23,13 +23,13 @@ import ( "strings" "time" - "github.com/apache/spark-connect-go/spark/client/base" + "github.com/caldempsey/spark-connect-go/spark/client/base" - "github.com/apache/spark-connect-go/spark/client/options" + "github.com/caldempsey/spark-connect-go/spark/client/options" "google.golang.org/grpc/metadata" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" "google.golang.org/grpc" "google.golang.org/grpc/codes" ) diff --git a/spark/client/retry_test.go b/spark/client/retry_test.go index a9526e5..83801db 100644 --- a/spark/client/retry_test.go +++ b/spark/client/retry_test.go @@ -22,11 +22,11 @@ import ( "testing" "time" - "github.com/apache/spark-connect-go/spark/client/options" + "github.com/caldempsey/spark-connect-go/spark/client/options" - "github.com/apache/spark-connect-go/spark/client/testutils" - "github.com/apache/spark-connect-go/spark/mocks" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/client/testutils" + "github.com/caldempsey/spark-connect-go/spark/mocks" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" "github.com/stretchr/testify/assert" "google.golang.org/grpc/codes" diff --git a/spark/client/testutils/utils.go b/spark/client/testutils/utils.go index c0b3bb5..e38a33b 100644 --- a/spark/client/testutils/utils.go +++ b/spark/client/testutils/utils.go @@ -19,7 +19,7 @@ import ( "context" "testing" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" "google.golang.org/grpc" ) diff --git a/spark/mocks/mock_executor.go b/spark/mocks/mock_executor.go index 600e9e0..fc4e9a6 100644 --- a/spark/mocks/mock_executor.go +++ b/spark/mocks/mock_executor.go @@ -19,13 +19,13 @@ import ( "context" "errors" - "github.com/apache/spark-connect-go/spark/sql/utils" + "github.com/caldempsey/spark-connect-go/spark/sql/utils" - "github.com/apache/spark-connect-go/spark/client/base" + "github.com/caldempsey/spark-connect-go/spark/client/base" "github.com/apache/arrow-go/v18/arrow" - "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sql/types" ) type TestExecutor struct { diff --git a/spark/mocks/mocks.go b/spark/mocks/mocks.go index 3a313f2..c6232a9 100644 --- a/spark/mocks/mocks.go +++ b/spark/mocks/mocks.go @@ -25,7 +25,7 @@ import ( "github.com/google/uuid" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" "google.golang.org/grpc/metadata" ) diff --git a/spark/sql/column/column.go b/spark/sql/column/column.go index 79e2fad..13eb6a8 100644 --- a/spark/sql/column/column.go +++ b/spark/sql/column/column.go @@ -18,9 +18,9 @@ package column import ( "context" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" ) // Convertible is the interface for all things that can be converted into a protobuf expression. diff --git a/spark/sql/column/column_test.go b/spark/sql/column/column_test.go index fa97e80..c62cbde 100644 --- a/spark/sql/column/column_test.go +++ b/spark/sql/column/column_test.go @@ -19,7 +19,7 @@ import ( "context" "testing" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/column/expressions.go b/spark/sql/column/expressions.go index 399a611..ca0bfcb 100644 --- a/spark/sql/column/expressions.go +++ b/spark/sql/column/expressions.go @@ -20,11 +20,11 @@ import ( "fmt" "strings" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" ) func newProtoExpression() *proto.Expression { diff --git a/spark/sql/column/expressions_test.go b/spark/sql/column/expressions_test.go index 836d5a8..4fb59e3 100644 --- a/spark/sql/column/expressions_test.go +++ b/spark/sql/column/expressions_test.go @@ -20,7 +20,7 @@ import ( "reflect" "testing" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/dataframe.go b/spark/sql/dataframe.go index dd8cb26..d602030 100644 --- a/spark/sql/dataframe.go +++ b/spark/sql/dataframe.go @@ -22,14 +22,14 @@ import ( "math/rand/v2" "github.com/apache/arrow-go/v18/arrow" - "github.com/apache/spark-connect-go/spark/sql/utils" + "github.com/caldempsey/spark-connect-go/spark/sql/utils" - "github.com/apache/spark-connect-go/spark/sql/column" + "github.com/caldempsey/spark-connect-go/spark/sql/column" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) // ResultCollector receives a stream of result rows diff --git a/spark/sql/dataframe_test.go b/spark/sql/dataframe_test.go index 0bee27b..49fb11f 100644 --- a/spark/sql/dataframe_test.go +++ b/spark/sql/dataframe_test.go @@ -20,8 +20,8 @@ import ( "context" "testing" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sql/functions" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sql/functions" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/dataframenafunctions.go b/spark/sql/dataframenafunctions.go index 9845bb1..23f827f 100644 --- a/spark/sql/dataframenafunctions.go +++ b/spark/sql/dataframenafunctions.go @@ -18,7 +18,7 @@ package sql import ( "context" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" ) type DataFrameNaFunctions interface { diff --git a/spark/sql/dataframewriter.go b/spark/sql/dataframewriter.go index 8c096f8..6f98649 100644 --- a/spark/sql/dataframewriter.go +++ b/spark/sql/dataframewriter.go @@ -21,8 +21,8 @@ import ( "fmt" "strings" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) // DataFrameWriter supports writing data frame to storage. diff --git a/spark/sql/dataframewriter_test.go b/spark/sql/dataframewriter_test.go index bc85f65..8a6775c 100644 --- a/spark/sql/dataframewriter_test.go +++ b/spark/sql/dataframewriter_test.go @@ -19,10 +19,10 @@ import ( "context" "testing" - "github.com/apache/spark-connect-go/spark/client" + "github.com/caldempsey/spark-connect-go/spark/client" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/mocks" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/mocks" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/functions/buiitins.go b/spark/sql/functions/buiitins.go index 4dca8bf..8f07d82 100644 --- a/spark/sql/functions/buiitins.go +++ b/spark/sql/functions/buiitins.go @@ -16,8 +16,8 @@ package functions import ( - "github.com/apache/spark-connect-go/spark/sql/column" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/column" + "github.com/caldempsey/spark-connect-go/spark/sql/types" ) func Expr(expr string) column.Column { diff --git a/spark/sql/functions/generated.go b/spark/sql/functions/generated.go index 844fa61..f66f9a7 100644 --- a/spark/sql/functions/generated.go +++ b/spark/sql/functions/generated.go @@ -15,7 +15,7 @@ package functions -import "github.com/apache/spark-connect-go/spark/sql/column" +import "github.com/caldempsey/spark-connect-go/spark/sql/column" // BitwiseNOT - Computes bitwise not. // diff --git a/spark/sql/group.go b/spark/sql/group.go index 3943969..125fee8 100644 --- a/spark/sql/group.go +++ b/spark/sql/group.go @@ -19,12 +19,12 @@ package sql import ( "context" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sparkerrors" - "github.com/apache/spark-connect-go/spark/sql/column" - "github.com/apache/spark-connect-go/spark/sql/functions" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/sql/column" + "github.com/caldempsey/spark-connect-go/spark/sql/functions" ) type GroupedData struct { diff --git a/spark/sql/group_test.go b/spark/sql/group_test.go index fe02b49..f6e6c72 100644 --- a/spark/sql/group_test.go +++ b/spark/sql/group_test.go @@ -19,10 +19,10 @@ import ( "context" "testing" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/client" - "github.com/apache/spark-connect-go/spark/client/testutils" - "github.com/apache/spark-connect-go/spark/mocks" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/client" + "github.com/caldempsey/spark-connect-go/spark/client/testutils" + "github.com/caldempsey/spark-connect-go/spark/mocks" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/plan.go b/spark/sql/plan.go index 022298f..89709ae 100644 --- a/spark/sql/plan.go +++ b/spark/sql/plan.go @@ -19,7 +19,7 @@ package sql import ( "sync/atomic" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" ) var atomicInt64 atomic.Int64 diff --git a/spark/sql/sparksession.go b/spark/sql/sparksession.go index a84bb61..433a1a7 100644 --- a/spark/sql/sparksession.go +++ b/spark/sql/sparksession.go @@ -23,19 +23,19 @@ import ( "time" "github.com/apache/arrow-go/v18/arrow/memory" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" "github.com/apache/arrow-go/v18/arrow" "github.com/apache/arrow-go/v18/arrow/array" "github.com/apache/arrow-go/v18/arrow/ipc" - "github.com/apache/spark-connect-go/spark/client/base" + "github.com/caldempsey/spark-connect-go/spark/client/base" - "github.com/apache/spark-connect-go/spark/client/options" + "github.com/caldempsey/spark-connect-go/spark/client/options" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/client" - "github.com/apache/spark-connect-go/spark/client/channel" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/client" + "github.com/caldempsey/spark-connect-go/spark/client/channel" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" "github.com/google/uuid" "google.golang.org/grpc/metadata" ) diff --git a/spark/sql/sparksession_integration_test.go b/spark/sql/sparksession_integration_test.go index c23d671..9512fd4 100644 --- a/spark/sql/sparksession_integration_test.go +++ b/spark/sql/sparksession_integration_test.go @@ -21,7 +21,7 @@ import ( "github.com/apache/arrow-go/v18/arrow" "github.com/apache/arrow-go/v18/arrow/memory" - "github.com/apache/spark-connect-go/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/sparksession_test.go b/spark/sql/sparksession_test.go index 11539af..9dedfae 100644 --- a/spark/sql/sparksession_test.go +++ b/spark/sql/sparksession_test.go @@ -29,11 +29,11 @@ import ( "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/client" - "github.com/apache/spark-connect-go/spark/client/testutils" - "github.com/apache/spark-connect-go/spark/mocks" - "github.com/apache/spark-connect-go/spark/sparkerrors" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/client" + "github.com/caldempsey/spark-connect-go/spark/client/testutils" + "github.com/caldempsey/spark-connect-go/spark/mocks" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) func TestSparkSessionTable(t *testing.T) { diff --git a/spark/sql/types/arrow.go b/spark/sql/types/arrow.go index 6c49bd0..0bfc1ff 100644 --- a/spark/sql/types/arrow.go +++ b/spark/sql/types/arrow.go @@ -20,13 +20,13 @@ import ( "bytes" "fmt" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" "github.com/apache/arrow-go/v18/arrow" "github.com/apache/arrow-go/v18/arrow/array" "github.com/apache/arrow-go/v18/arrow/ipc" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) func ReadArrowTableToRows(table arrow.Table) ([]Row, error) { diff --git a/spark/sql/types/arrow_test.go b/spark/sql/types/arrow_test.go index 92d491b..6d88c66 100644 --- a/spark/sql/types/arrow_test.go +++ b/spark/sql/types/arrow_test.go @@ -30,8 +30,8 @@ import ( "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sql/types" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sql/types" ) func TestShowArrowBatchData(t *testing.T) { diff --git a/spark/sql/types/builtin.go b/spark/sql/types/builtin.go index 1f74695..944beca 100644 --- a/spark/sql/types/builtin.go +++ b/spark/sql/types/builtin.go @@ -19,7 +19,7 @@ package types import ( "context" - proto "github.com/apache/spark-connect-go/internal/generated" + proto "github.com/caldempsey/spark-connect-go/internal/generated" ) type LiteralType interface { diff --git a/spark/sql/types/conversion.go b/spark/sql/types/conversion.go index b2652e2..92a2fc5 100644 --- a/spark/sql/types/conversion.go +++ b/spark/sql/types/conversion.go @@ -19,8 +19,8 @@ package types import ( "errors" - "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sparkerrors" + "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sparkerrors" ) func ConvertProtoDataTypeToStructType(input *generated.DataType) (*StructType, error) { diff --git a/spark/sql/types/conversion_test.go b/spark/sql/types/conversion_test.go index cd62779..109dc92 100644 --- a/spark/sql/types/conversion_test.go +++ b/spark/sql/types/conversion_test.go @@ -19,8 +19,8 @@ package types_test import ( "testing" - proto "github.com/apache/spark-connect-go/internal/generated" - "github.com/apache/spark-connect-go/spark/sql/types" + proto "github.com/caldempsey/spark-connect-go/internal/generated" + "github.com/caldempsey/spark-connect-go/spark/sql/types" "github.com/stretchr/testify/assert" ) diff --git a/spark/sql/types/rowiterator_test.go b/spark/sql/types/rowiterator_test.go index 99672de..1567ccd 100644 --- a/spark/sql/types/rowiterator_test.go +++ b/spark/sql/types/rowiterator_test.go @@ -13,7 +13,7 @@ import ( "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" - "github.com/apache/spark-connect-go/v40/spark/sql/types" + "github.com/caldempsey/spark-connect-go/spark/sql/types" ) func TestRowIterator_BasicIteration(t *testing.T) { diff --git a/spark/sql/utils/consts.go b/spark/sql/utils/consts.go index e1312ef..3dd5f61 100644 --- a/spark/sql/utils/consts.go +++ b/spark/sql/utils/consts.go @@ -15,7 +15,7 @@ package utils -import proto "github.com/apache/spark-connect-go/internal/generated" +import proto "github.com/caldempsey/spark-connect-go/internal/generated" type ExplainMode int