Fix Test log inputs with large inputs limit check

jgiannuzzi/mlflow-go#47

Signed-off-by: nojaf <florian.verdonck@outlook.com>

.github/workflows/test.yml

@@ -56,4 +56,4 @@ jobs:
       - name: Run integration tests
         run: mage test:python
         # Temporary workaround for failing tests
-        continue-on-error: true
+        continue-on-error: ${{ matrix.runner != 'ubuntu-latest' }}

pkg/entities/dataset.go

@@ -1,6 +1,8 @@
 package entities
 
-import "github.com/mlflow/mlflow-go/pkg/protos"
+import (
+	"github.com/mlflow/mlflow-go/pkg/protos"
+)
 
 type Dataset struct {
 	Name       string
@@ -12,12 +14,22 @@ type Dataset struct {
 }
 
 func (d *Dataset) ToProto() *protos.Dataset {
+	var schema *string
+	if d.Schema != "" {
+		schema = &d.Schema
+	}
+
+	var profile *string
+	if d.Profile != "" {
+		profile = &d.Profile
+	}
+
 	return &protos.Dataset{
 		Name:       &d.Name,
 		Digest:     &d.Digest,
 		SourceType: &d.SourceType,
 		Source:     &d.Source,
-		Schema:     &d.Schema,
-		Profile:    &d.Profile,
+		Schema:     schema,
+		Profile:    profile,
 	}
 }